Pytorch——报错解决:No inf checks were recorded for this optimizer.
发布时间:2023-08-21 20:58:10 1102
相关标签:
报错截图
报错原因
- 我这里报错是因为loss中出现了nan值,所以导致该问题。
解决方法
- 直接做法: 给loss最终的结果加上
torch.nan_to_num()
函数,可以避免nan值。 - 但是最好应当查明为什么出现nan值,针对性解决才能真正解决bug。
参考文章:
- AssertionError:在Pytorch的AutomaticMixedPrecision中未记录此优化器的inf检查
文章来源: https://blog.51cto.com/u_15906550/5921412
特别声明:以上内容(图片及文字)均为互联网收集或者用户上传发布,本站仅提供信息存储服务!如有侵权或有涉及法律问题请联系我们。
举报