回归任务中神经网络训练loss为nan

Remote Sensing

1429人浏览 · 2020-05-24 11:19:42

Remote Sensing · 2020-05-24 11:19:42 发布

问题

我最近参加数学建模比赛，写了一个三层的全连接神经网络，用于回归任务，但是喂入数据后发现loss从第一个epoch就是nan，训练无法进行。

解决方法

只对样本X进行了标准化，而Y没进行标准化。从而导致整体的loss极大，最终成为nan。因此只需要对Y也进行相同的标准化。

之前还遇到过一次，是由于数据根本没能喂入网络造成，也是原因之一。

魔乐社区

魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。

更多推荐

小参数・大码力・易部署 | Qwen3.6-27B上线魔乐社区，基于昇腾的部署教程来了

继一周前模型开源发布后，千问再度开源Qwen3.6-27B —— 一个拥有270亿参数的稠密多模态模型，也是社区呼声最高的模型规格。Qwen3.6-27B 依然支持多模态思考与非思考模式，在智能体编程方面达到了旗舰级表现，全面超越前代开源旗舰 Qwen3.5-397B-A17B（总参数397B / 激活参数17B的MoE模型）。作为稠密架构，它无需MoE路由即可部署，是开发者在实用、可广泛部署规模