Thanks, I updated the pytorch to 1.7, the result of the prediction is still out of [-1,1] print(np.unique(pred))
[-38.52738  -38.197113 -37.780617 ...  10.451873  10.506324  10.691366]

Check of the pytorch version (was installed via pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html)
torch.__version__

Out[2]:

'1.7.1+cu101'