Runtime Error re: dimensions

[andrewsx]

Hello,
I am getting this error when running your train.py file. Do you have any suggestions on why that might be?

RuntimeError: Expected 4-dimensional input for 4-dimensional weight 64 1 3, but got 3-dimensional input of size [128, 32, 32] instead

[pmorye]

This is because the input that we are providing for forward pass is a batch of images (batch size = 128) and which have only 1 channel (the luminescence channel) which we are not specifiying here. To correct this, we need to change the shape of the input l_images to [128, 1, 32, 32] meaning that each image has 1 channel (By doing: l_images = l_images.unsqueeze(1)).