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