使用checkpoint时,警告使用checkpoint时,警告UserWarning: None of the inputs have requires_grad=True.
原因:使用checkpoint, 不能放在第一个位置, 或者说放在第一个就要传入一个有梯度的, 不然就无法进行反向传播.
原程序:
更改为:

即可正常运行
在使用PyTorch的checkpoint功能时,遇到UserWarning提示没有输入变量需要求梯度。问题在于checkpoint应跟随至少一个有梯度的输入。原程序中checkpoint被错误地放在了首位,导致无法进行反向传播。解决方法是确保checkpoint之后的参数至少有一个具有requires_grad=True的属性,以允许计算梯度。
使用checkpoint时,警告使用checkpoint时,警告UserWarning: None of the inputs have requires_grad=True.
原因:使用checkpoint, 不能放在第一个位置, 或者说放在第一个就要传入一个有梯度的, 不然就无法进行反向传播.
原程序:
更改为:

即可正常运行
524
8786
440

被折叠的 条评论
为什么被折叠?