报错截图
报错原因我这里报错是因为loss中出现了nan值,所以导致该问题。解决方法直接做法: 给loss最终的结果加上??torch.nan_to_num()??函数,可以避免nan值。但是最好应当查明为什么出现nan值,针对性解决才能真正解决bug。
参考文章:
??AssertionError:在Pytorch的AutomaticMixedPrecision中未记录此优化器的inf检查??
【转自:韩国服务器 http://www.yidunidc.com 欢迎留下您的宝贵建议】刺是与生俱来的,上帝在赐予优越感同时捆-绑的附属品;