Tensors shapes do not correspond in PyTorch
I have designed a network for instance segmentation and every time I load the mask files, it gives me an error by saying that the shapes do not correspond. The loaded masks' tensor shape is (32, 224, 224) (with the following order NxHxW).