CODEWITHSUNDEEP
machine-learningneural-networkpytorchopenai-gym
Gulzar
Gulzar

Reputation: 27966

How to look at the parameters of a pytorch model?

I have a simple pytorch neural net that I copied from openai, and I modified it to some extent (mostly the input).

When I run my code, the output of the network remains the same on every episode, as if no training occurs.

I want to see if any training happens, or if some other reason causes the results to be the same.

How can I make sure any movement happens to the weights?

Thanks

Upvotes: 1

Views: 2157

Answers (1)

Szymon Maszke
Szymon Maszke

Reputation: 24701

Depends on what you are doing, but the easiest would be to check the weights of your model.

You can do this (and compare with the ones from previous iteration) using the following code:

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

If the weights are changing, the neural network is being optimized (which doesn't necessarily mean it learns anything useful in particular).

Upvotes: 4

Related Questions