How to use TensorBoard with PyTorch — PyTorch Tutorials 1.10 ...
pytorch.org › tutorials › recipesHow to use TensorBoard with PyTorch. TensorBoard is a visualization toolkit for machine learning experimentation. TensorBoard allows tracking and visualizing metrics such as loss and accuracy, visualizing the model graph, viewing histograms, displaying images and much more. In this tutorial we are going to cover TensorBoard installation, basic usage with PyTorch, and how to visualize data you logged in TensorBoard UI.