解决办法:
eps 改为 1e-4
(默认的1e-8在half下会round成0,除以0时会出现Nan)
本文标题:pytroch 模型设置成half后,使用Adam,loss出现
本文链接:https://www.haomeiwen.com/subject/ciyucctx.html
网友评论