如何查看pytorch模型的参数?

问题描述 投票:0回答:1

我有一个简单的pytorch神经网络,我从openai复制,我在某种程度上修改它(主要是输入)。

当我运行我的代码时,每个剧集的网络输出都保持不变,就像没有训练一样。

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

如何确保重量发生任何移动?

谢谢

machine-learning neural-network pytorch openai-gym
1个回答
3
投票

取决于你在做什么,但最简单的方法是检查模型的重量。

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

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

如果权重正在改变,神经网络正在被优化(这并不一定意味着它学习了特别有用的东西)。

© www.soinside.com 2019 - 2024. All rights reserved.