Tutorial: Tensorboard in Pytorch
Last updated
Was this helpful?
Last updated
Was this helpful?
In this tutorial, we implement a MNIST classifier using a simple neural network and visualize the training process using . In training phase, we plot the loss and accuracy functions through scalar_summary
and visualize the training images through image_summary
. In addition, we visualize the weight and gradient values of the parameters of the neural network using histogram_summary
. PyTorch code for handling these summary functions can be found .
To run the TensorBoard, open a new terminal and run the command below. Then, open on your web browser.