问题: 在训练的时候用了两次loss.backward,然后就报错了。
解决办法在backward()
函数中添加参数retain_graph=True
:
loss.backward()
改为
loss.backward(retain_graph=True)
问题: 在训练的时候用了两次loss.backward,然后就报错了。
解决办法在backward()
函数中添加参数retain_graph=True
:
loss.backward()
改为
loss.backward(retain_graph=True)