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: CompositionalMetric) → LossProtocol[Tensor, Tensor]

Returns a wrapper of a CompositionalMetric for integration.

Parameters:

torch_metric (CompositionalMetric)

Return type:

LossProtocol[Tensor, Tensor]