I using a pre-trained resnet18 for my application. But I’m getting the above error. The input image size is [torch.FloatTensor of size 2x75x75]. I have attached the error screenshot below
transferConv.ipynb.jpg978×836 239 KB
. Can anyone point out a possible solution for this problem? Thank you.