CODEWITHSUNDEEP
matplotlibpytorchgraph-visualization
user11140802
user11140802

Reputation:

How to visualize my training history in pytorch?

How do you guys visualize the training history of your pytorch model like in keras here.

I have a pytorch trained model and I want to see the graph of its training. Can I do this using only matplotlib? If yes, can someone give me resources to follow.

Upvotes: 3

Views: 7403

Answers (1)

Abhishek Verma
Abhishek Verma

Reputation: 1729

You have to save the loss while training. A trained model won't have history of its loss. You need to train again.

Save the loss while training then plot it against the epochs using matplotlib. In your training function, where loss is being calculated save that to a file and visualize it later. Also, you can use tensorboardX if you want to visualize in realtime.

This is a tutorial for tensorboardX: http://www.erogol.com/use-tensorboard-pytorch/

Upvotes: 1

Related Questions