yes, I have five channels in my images (batch_size, 5, 64, 64) because they are coming from sentinel satellite.



 ptrblck:

batch_correct = (label.eq(pred.long())).double().sum().item()


Actually I have already used double() before because I read on this forum that equating pytorch tensors return a byte tensor so it is a good idea to cast them to double. But it still doesn’t work.
Just checked it now with batch size 1 (98.31%)

and batch size 512 (94.26%).