回归任务中神经网络训练loss为nan
问题我最近参加数学建模比赛，写了一个三层的全连接神经网络，用于回归任务，但是喂入数据后发现loss从第一个epoch就是nan，训练无法进行。解决方法只对样本X进行了标准化，而Y没进行标准化。从而导致整体的loss极大，最终成为nan。因此只需要对Y也进行相同的标准化。之前还遇到过一次，是由于数据根本没能喂入网络造成，也是原因之一。...
·
技术共进，成长同行——讯飞AI开发者社区
更多推荐
1
0- 0
已为社区贡献4条内容
所有评论(0)