How to Plot accuracy and loss graph after training completion?

I am training a CNN through transfer learning, implemented the basic provided tutorial by pytorch (Transfer Learning for Computer Vision Tutorial — PyTorch Tutorials 1.7.1 documentation). I wanna plot accuracy and loss graph, but not getting getting any idea about it. I shall be grateful, if anybody help me out in this regard.

TensorboardX will do the work for you.

1 Like

Use Tensorboard or save each epoch’s loss and accuracy to a list and plot it with matplotlib.

I used list method, it worked. Many Thanks :slight_smile: