keras踩坑记——gpu和cpu结果不同

问题描述: 训练了大概6个epoch 模型在cpu上的loss开始出现nan,在gpu上表现正常


经查找资料https://www.oschina.net/question/3683714_2266089 这位兄台和我遇到了同样的问题,然而他并没有解决


https://www.zhihu.com/question/62441748

这位兄台给出了一个解释,但是和我的模型问题应该不相同,因为我的RNN模型的换了激活函数(relu-->sigmoid)后表现正常,因此应该不是标签的问题


问题仍未解决,恳请各位大佬不吝赐教

猜你喜欢

转载自blog.csdn.net/huowa9077/article/details/80236849
今日推荐