使用PyTorch常见4个错误解决示例详解

来自:网络
时间:2022-12-28
阅读:
目录

导读

这4个错误,我敢说大部分人都犯过,希望能给大家一点提醒。

使用PyTorch常见4个错误解决示例详解

最常见的神经网络错误:

1)你没有首先尝试过拟合单个batch。

2)你忘了为网络设置train/eval模式。

3)在.backward()之前忘记了.zero_grad()(在pytorch中)。

4)将softmaxed输出传递给了期望原始logits的损失,还有其他吗?

返回顶部
顶部