Code

Allow metrics layers to have state.#978

Code

Merged

markblee merged 2 commits intoapple:mainapple/axlearn:mainfrom markblee:markblee/metrics_base_layermarkblee/axlearn:markblee/metrics_base_layerCopy head branch name to clipboardFeb 11, 2025

Commits

Commits on Feb 10, 2025

• Allow metrics layers to have state.

  markblee

  committed
  bc3b1ff

  Copy full SHA for bc3b1ff

Commits on Feb 11, 2025

• Move BaseLossMetrics to a new file.

  markblee

  committed
  3c6abe5

  Copy full SHA for 3c6abe5