results are broken

[pumpkinsoup0]

Hello,

I am writing to you because I am interested in using your code for my undergraduate graduation thesis. I have tested the code, but the results are as follows. I am not sure what the problem is. My environment is Ubuntu 22.04 / Pytorch 1.1.0 / cudatoolkit 10.0. There is no problem with using the GPU, but the results are broken.

I would be grateful if you could help me to identify the problem.

Thank you for your time and consideration.

[pumpkinsoup0]

it worked well at Pytorch 1.7.1 / cudatoolkit 11.0