美文网首页
解决RuntimeError: one of the varia

解决RuntimeError: one of the varia

作者: JIAWEIJIAWEI | 来源:发表于2022-09-22 17:07 被阅读0次

    需要更新的参数被原地操作更改了
    可能是以下几种情况导致
    1)找到网络模型中的 inplace 操作,将inplace=True改成 inplace=False,例如torch.nn.ReLU(inplace=False)

    2)将代码中的“a+=b”之类的操作改为“c = a + b”,a=b改成a=ab,a/=b改成a=a/b。注意认真找。

    3)训练代码的optimizer.step()函数放到loss.backward()后面

    4)pytorch降版本到1.4或1.5

    相关文章

      网友评论

          本文标题:解决RuntimeError: one of the varia

          本文链接:https://www.haomeiwen.com/subject/atowortx.html