Yeah that was totally my bad.
I changed MSELoss to L1Loss at the same time as I removed a .float() somewhere.
L! does not seam to care, MSE does.
Also at the same time I tried to use MSLELoss…
Yeah I should go to sleep at 5am.
thanks for showing me the obvious!
Guys: go to sleep and try again the next day before making forum posts