Hy!
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