This repository shows how to use TensorBoard from PyTorch with MNIST example.
tensorboard-pytorch (tensorboardX), a TensorBoard extension for PyTorch, is used for this.
pip install -r requirements.txt
python main.py # start streaming data to the TensorBoard while training the model
# In another terminal,
tensorboard --logdir runs
# Then open localhost:6006 in your browser, and you will see something like the screenshots below.