当前位置:   article > 正文

使用checkpoint遇到的问题_v1-5-pruned-emaonly

v1-5-pruned-emaonly

使用checkpoint时,警告使用checkpoint时,警告UserWarning: None of the inputs have requires_grad=True. 

原因:使用checkpoint, 不能放在第一个位置, 或者说放在第一个就要传入一个有梯度的, 不然就无法进行反向传播.

原程序: 

 

更改为:

 

即可正常运行 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/585440
推荐阅读
相关标签
  

闽ICP备14008679号