我的代码中是什么导致我在 Pytorch 中训练网络时出现无限错误？

Question

这是我的代码：

我正在处理一个非常大的数据集（有几百万个值）。我的输入数据有 12 个特征，而我的输出数据有 1 个特征。这是我第一次使用 Pytorch，所以我从基本的线性网络开始。当我运行几个时期时，我得到了无限错误，如你所见。

编辑（更新我的代码以使用 DataLoader）：

在此输入图片描述

截至 18 个 epoch，我仍然有 nan 错误，我不知道为什么。

Answer 1

zero_grad和backward之间的位置错误，您应该先放置backward，然后应用zero_grad