0
点赞
收藏
分享

微信扫一扫

Pytorch——报错解决:No inf checks were recorded for this optimizer.


报错截图

Pytorch——报错解决:No inf checks were recorded for this optimizer._解决bug

报错原因

  • 我这里报错是因为loss中出现了nan值,所以导致该问题。

解决方法

  • 直接做法: 给loss最终的结果加上​​torch.nan_to_num()​​函数,可以避免nan值。
  • 但是最好应当查明为什么出现nan值,针对性解决才能真正解决bug。

参考文章:

  • ​​AssertionError:在Pytorch的AutomaticMixedPrecision中未记录此优化器的inf检查​​


举报

相关推荐

0 条评论