`latte.metrics.torch.wrapper`

Module Contents

Classes

TorchMetricWrapper

Functions

`torch_to_numpy`(args, kwargs)
`numpy_to_torch`(val)

latte.metrics.torch.wrapper.torch_to_numpy(args, kwargs)

latte.metrics.torch.wrapper.numpy_to_torch(val)

class latte.metrics.torch.wrapper.TorchMetricWrapper(metric: Callable[Ellipsis, latte.metrics.base.LatteMetric], name: Optional[str] = None, compute_on_step: bool = False, dist_sync_on_step: bool = False, process_group: Optional[Any] = None, dist_sync_fn: Callable = None, **kwargs)

Bases: torchmetrics.Metric

update(self, *args, **kwargs)

compute(self)

reset(self)