Hi,
I have an annoying issue that I can’t handle.
Here is a reproducible code in colab :
https://colab.research.google.com/drive/1LixjriTogEgCs_zPRnNgoXb-jxEBOuYw 4
It appears in the Running train-loop session,
First of all I have this line :
1/5 * Epoch (train): 0% 0/1  , which is quite weird, something like 0/1.
Then I have the classical issue :  The size of tensor a must match the size of tensor b at non-singleton dimension 4 , that I can’t understand as in pytorch everything is moreor less hidden, I don’t know where it could come from.
I’m quite new in Pytorch, may you have an idea ?
Thanks