Please for help.
I trained the model, which was constructed in Pytorch.
But there is a error. It shows in the figure.

problem.JPG1027×451 63.6 KB

The environment is:
Ubuntu 16.04
Python 3.6
Pytorch 1.1.0
My GPU is GeForce GTX 1080.