Closed alan-cooney closed 7 months ago
Switch all metrics and loss to extend torchmetrics. This will allow them to work across processes with DDP.
Switch all metrics and loss to extend torchmetrics. This will allow them to work across processes with DDP.