码迷,mamicode.com
首页 > 其他好文 > 详细

转:sqrt()可能会使神经网络loss变为NaN

时间:2020-09-17 23:12:38      阅读:29      评论:0      收藏:0      [点我收藏+]

标签:tor   end   none   ref   神经网络   arc   出现   hub   网络   

pytorch loss 出现nan,原因之一,计算图中存在torch.sqrt

loss突然变为nan?你可能踩了sqrt()的坑

 

转:sqrt()可能会使神经网络loss变为NaN

标签:tor   end   none   ref   神经网络   arc   出现   hub   网络   

原文地址:https://www.cnblogs.com/picassooo/p/13657767.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!