pytorch训练 loss=inf或者训练过程中loss=Nan
造成 loss=inf的原因之一:data underflow最近在测试Giou的测试效果,在mobilenetssd上面测试Giou loss相对smoothl1的效果;改完后训练出现loss=inf原因: 在使用log函数时出现 data underflow解决方法:增加一个bias原代码 # match wh / prior wh g_wh = (matched[:, 2:] - matched[:, :2]) / priors[:, 2:] g_wh = torc
原创
2021-03-30 20:50:55 ·
7789 阅读 ·
0 评论