I had a look and printed the batch.size()
OUT:
torch.Size([64, 1, 224, 224])
Yes so the transform was to resize the images to 224 by 224 as they are originally 64x64.
Not really sure why it’s producing an error now