MetricModule#

class MetricModule(tag, metrics)[source]#

Bases: Module

Module for metrics. It’s a wrapper of MetricCollections for training, validation and test stage. It’s used in BaseHead class. In general, it’s not used directly by users.

Methods

compute

forward

reset

update

Attributes

Parameters:

  • tag (str) –

  • metrics (Metric | list[torchmetrics.metric.Metric] | dict[str, torchmetrics.metric.Metric]) –