feat: extend EpochMetric to support more compute_fn output types - #3789

Open

zongyang078 wants to merge 9 commits into

pytorch:masterfrom

zongyang078:feature/epoch-metric-output-types

Commits on Jun 9, 2026

feat: extend EpochMetric to support more compute_fn output types
zongyang078
committed
test: add tests for EpochMetric with tensor, tuple and invalid output types
zongyang078
committed

Commits on Jun 10, 2026

docs: update EpochMetric docstring to document supported compute_fn output types
zongyang078
committed

Commits on Jun 19, 2026

Merge remote-tracking branch 'upstream/master' into feature/epoch-metric-output-types
zongyang078
committed

Commits on Jun 24, 2026

refactor: use EpochMetricOutput type alias and add list/dict output tests
zongyang078
committed

Commits on Jul 2, 2026

Commits on Jul 8, 2026