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]) –