训练完之后加载训练好的模型一直报nan

解决方法： 

（1）之前设的gpu为1，batch size为1    后边改为gpu为2，batch size为8

 

 

   

 

 结果为

 设置的参数影响训练的模型

MAX_ITER改为90000的结果：

 

posted @ 2021-07-11 20:45  赵立敏  阅读(89)  评论(0)  编辑  收藏  举报