參考：https://blog.csdn.net/chen645096127/article/details/94019443
報錯原因分析:
train_loss += loss.data[0] 是pytorch0.3.1版本代碼,在0.4-0.5版本的pytorch會出現(xiàn)警告,不會報錯,但是0.5版本以上的pytorch就會報錯,總的來說是版本更新問題.
解決方法:
#將原語句：
train_loss+=loss.data[0]
#修改為：
train_loss+=loss.item()
#bingo
#原出錯語句飞傀，在torch-0.4.0上運(yùn)行時未出錯蚓炬，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() ))
