I tried to fine tune the ResNe18 model on GPU. The training code runs OK. However, I decided to further train the produced model using a different loss function (switched from L1 loss to L2 loss) but got the the error below. No other code or data changed.
RuntimeError: Found dtype Double but expected Float
Any idea what might have caused the issue and how to fix it?
Below is the information of my environment.
OS: Ubuntu 20.04 in Docker python 3.8.10 torch 1.9.1+cu111 torchaudio 0.9.1 torchvision 0.10.1+cu111