Does the project code have to rely on pytorch 0.4? Is it possible to use a higher version of pytorch?

[NGC2237-MyRose]

I find it difficult to install such an old version of pytorch on linux today, can I use another version of pytorch?

[KikkiZ]

I ran the code in pytorch 2.1 and python 3.9, but I changed some of the code. I mean, you can run the code in the pytorch environment that you have, there may be some incompatibilities, you just need to make a few changes.

[zcy5417]

已收到！祝每天开开心心！！