如何查看pytorch模型的参数？

Question

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

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

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

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

谢谢

Answer 1

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

您可以使用以下代码执行此操作（并与之前迭代的代码进行比较）：

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

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