Ada Puślecka
08/25/2023, 12:19 PMtorchmetrics.metric.Metric
and thinking if it would be possible to use that.Cristian (Nixtla)
08/25/2023, 1:42 PMBasePointLoss
to facilitate adding more losses. Here is the MSE
loss based on the new class.mask
. Let me know if you have particular doubts on how to create your loss.