machine-learning - 如何查看pytorch模型的参数？

Question

我有一个简单的 pytorch 神经网络，是从 openai 复制的。 ，并且我对其进行了一定程度的修改(主要是输入)。

当我运行代码时，网络的输出在每一集中都保持相同，就好像没有进行训练一样。

我想看看是否进行了任何训练，或者是否有其他原因导致结果相同。

如何确保重物发生任何移动？

谢谢

Answer 1

取决于您在做什么，但最简单的方法是检查模型的权重。

您可以使用以下代码来执行此操作(并与之前迭代中的进行比较):

for parameter in model.parameters():
    print(parameter.data)

如果权重发生变化，则神经网络正在优化(这并不一定意味着它学到了任何特别有用的东西)。