CommonLoopUtils icon indicating copy to clipboard operation
CommonLoopUtils copied to clipboard
verified publisher icon google

Metric and Collection inherit from PyTreeNode

Open cgarciae opened this issue 3 years ago • 0 comments

This PR makes Metric and Collection inherit from flax.struct.PyTreeNode, this makes creating new metric types more ergonomic as you no longer need to decorate the class with flax.struct.dataclass.

cgarciae avatar Jan 03 '23 18:01 cgarciae