pytorch 训练第一个epoch 中训练几个batch出现NAN的问题解决

问题如标题，发现是batch size 太大了，可能是因为有些数据放不到GPU上出错，pytorch又没有及时报错。建议把batch size 调小试试

posted @ 2022-03-07 11:18  鲜衣  阅读(233)  评论(0)  编辑  收藏  举报