Hands-On TensorBoard for PyTorch Developers

Video description

TensorBoard is a visualization library for TensorFlow that plots training runs, tensors, and graphs. TensorBoard has been natively supported since the PyTorch 1.1 release. In this course, you will learn how to perform Machine Learning visualization in PyTorch via TensorBoard. This course is full of practical, hands-on examples. You will begin with a quick introduction to TensorBoard and how it is used to plot your PyTorch training models. You will learn how to write TensorBoard events and run TensorBoard with PyTorch to obtain visualizations of the training progress of a neural network. You will visualize scalar values, images, text and more, and save them as events. You will log events in PyTorch–for example, scalar, image, audio, histogram, text, embedding, and back-propagation.

By the end of the course, you will be confident enough to use TensorBoard visualizations in PyTorch for your real-world projects.

What You Will Learn

  • Demonstrate TensorBoard visualizations with PyTorch models, including training curves, data distributions, data histograms, model graphs, and text embeddings
  • Log multiple parameters and events in PyTorch and easily use them for TensorBoard visualizations
  • Visualize numerous data types including scalar, vector, text, image, and audio data
  • View data and text embeddings in 2D and 3D
  • Use TensorBoard to detect errors and fix models with hands-on examples in Machine Learning, image classification, and NLP
  • Track and optimize hyperparameter tuning so you can display model configurations and measure performance to compare multiple models and reproduce experiments
  • Log events from PyTorch with a few lines of code


This course targets developers, data scientists, analysts, and AI/ML engineers who work with PyTorch and want to leverage the power of the TensorBoard library to visualize the training progress of their neural networks.

Requirement: This course requires basic familiarity with Python and an IDE (Jupyter Notebooks or Colab), together with basic familiarity with PyTorch for testing and training neural networks.

About The Author

Joe Papa: Joe Papa has an MSEE and over 23 years' experience in engineering R&D. He has led AI teams and developed Deep Learning models at Booz Allen and Perspecta Labs. Joe is also the founder of Mentorship.ai and has mentored hundreds of data scientists in Machine Learning, Deep Learning, and AI. He has taught over 6,000 students on Udemy in programming courses such as MATLAB.

Product information

  • Title: Hands-On TensorBoard for PyTorch Developers
  • Author(s): Joe Papa
  • Release date: March 2020
  • Publisher(s): Packt Publishing
  • ISBN: 9781838983604