参考:https://blog.csdn.net/chen645096127/article/details/94019443
报错原因分析:
train_loss += loss.data[0] 是pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错,总的来说是版本更新问题.
解决方法:
#将原语句:
train_loss+=loss.data[0]
#修改为:
train_loss+=loss.item()
#bingo
#原出错语句,在torch-0.4.0上运行时未出错,torch-1.13.1出错
self.total_step , d_loss_real.data[0],
self.G.attn1.gamma.mean().data[0], self.G.attn2.gamma.mean().data[0] ))
#改为
self.total_step , d_loss_real.item(),
self.G.attn1.gamma.mean().item(), self.G.attn2.gamma.mean().item() ))