此次实验的环境为pytorch=1.4.0。这个错误的原因在于pytorch的自动求导机制,在求loss的时候,由于我的loss写成了
loss = loss_0 + loss_1
这样可能可能对pytorch的自动求导机制造成了迷惑,于是我将上面一句话改成下述代码后,程序就可以正常执行了,但是具体的效果是否和上面等同,目前还不确定。
loss = loss_0.backward() + loss_1.backward()
loss = loss_0 + loss_1
loss = loss_0.backward() + loss_1.backward()