【已解决】运行GAN时Torch报错

目录

报错内容：

报错原因：

解决办法：

注意事项：

报错内容：

CPU下运行GAN代码报错

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [2048]] is at version 4; expected version 3 instead. Hint: enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).

报错原因：

torch高于1.4.0以后，梯度更新变了。算G的loss的计算图是包含D的，但是在G的backward之前更新了D的值，这时候再去计算就不是和forward时候对应的梯度了。