I am getting this error on fastai any help?
Target size (torch.Size([32])) must be the same as input size (torch.Size([64]))
I think the output of the network is of size 64, whereas the original output is of size 32. check the network and class labels again.
Thank you for the reply, but when i change my input to 64 then it throws an error
Target size (torch.Size([64])) must be the same as input size (torch.Size([128]))
It keep on doubling. What loss function are we supposed to use because it seems like its a loss function error. I am using BCEWithLogitLoss().
I think the output of the network is of size 64, whereas the original output is of size 32. check the network and class labels again.
Thank you for the reply, but when i change my input to 64 then it throws an error
Target size (torch.Size([64])) must be the same as input size (torch.Size([128]))
It keep on doubling. What loss function are we supposed to use because it seems like its a loss function error. I am using BCEWithLogitLoss().