Here is one way to visualize your model’s training process using tensorboard with Pytorch
#Inside Code for Training
from torch.utils.tensorboard import SummaryWriter()
tb = SummaryWriter()
#Add Metrics you want to evaluate
tb.add_scalar('Loss',loss,epoch)
tb.add_scalar('Accuracy',accuracy,epoch)
tb.add_scalar('F1Score',fa_score,epoch)
To start a tensorboard server go to the terminal and type
tensorboard --logdir=runs
