when I Remove the metric list the error is removed but I can’t show any metric during running.
the error comes from : intersection = torch.sum(gt * pr)
the size of gt is 8x512x512
the size of pr is 8x5x512x512
where the 8 is batch size and 5 is the class number
how can i solve this issu ?