使用PyTorch时，最常见的4个错误

导读

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

最常见的神经网络错误：1)你没有首先尝试过拟合单个batch。2)你忘了为网络设置train/eval模式。3)在.backward()之前忘记了.zero_grad()(在pytorch中)。4)将softmaxed输出传递给了期望原始logits的损失，还有其他吗？

批量计算 pytorch

0 人点赞