Sizes of tensors must match except in dimension 0

image
image


I tried the method in the forum but it didn’t solve it. When I use randomresizecrop, it doesn’t go wrong. When I use tramsform.resize, I get this error. How to fix it? thank you very much

Could you try to pass the size argument as a tuple: transforms.Resize((IMG_SIZE, IMG_SIZE))?

1 Like

I have same issue , I have also resolve by transform.Resize((Img_size,img_size))
be careful it is a tuple.

thank a lot:grinning:

thank a lot:grinning: