Higher pytorch version raise error when training with GAN Net

3

the whole code is here GitHub - lcy0604/EraseNet

can u take a few time to help me?