drytorch.contrib.torchmetrics

Module containing utilies to ensure compatibility with torchmetrics.

Functions

from_torchmetrics(torch_metric)

Returns a wrapper of a CompositionalMetric for integration.
from_torchmetrics(torch_metric: metric.CompositionalMetric) → p.LossProtocol[_Tensor, _Tensor][source]

Returns a wrapper of a CompositionalMetric for integration.

Parameters:

torch_metric (metric.CompositionalMetric)

Return type:

p.LossProtocol[_Tensor, _Tensor]