bug记录
下载别人代码后跑训练脚本出现一下问题
ValueError: can't optimize a non-leaf Tensor
在将参数传入Adam优化器时,不能优化非叶节点,但是该节点require_grad=true。
解决办法:判断一下是否为非叶节点,如果不是,则将require_grad设置为false
传入Adam后再将其设置为true
bug记录
下载别人代码后跑训练脚本出现一下问题
ValueError: can't optimize a non-leaf Tensor
在将参数传入Adam优化器时,不能优化非叶节点,但是该节点require_grad=true。
解决办法:判断一下是否为非叶节点,如果不是,则将require_grad设置为false
传入Adam后再将其设置为true