sManohar201:

Got it, i used the Transform functionality to rescale the images. Thank you.


Torch reads images as (batch_size , channel, width, height), so if you want to feed in one image at a time, add the line
 t.unsqueeze(0)

This will make the image 1X2X75X75 i.e. one image with 2 channels of 75X75 each.