Sanjayvarma11:

unsqueeze(1)


This is the error I get with unsqueeze(1) Thanks for your reply
RuntimeError: Given groups=1, weight of size [64, 3, 7, 7], expected input[8, 1, 600, 800] to have 3 channels, but got 1 channels instead
I don’t want to change the code of resnet.