TensorBoard
Discover how to integrate and run TensorBoard inside Jupyter notebooks to visualize PyTorch training processes. Learn to set up the environment, manage log directories, and use TensorBoard to track and analyze your model's performance during development.
We'll cover the following...
We'll cover the following...
TensorBoard introduction
TensorBoard is a very useful visualization tool from PyTorch’s competing framework, Tensorflow. And you can use this with PyTorch as well, which provides classes and methods for us to integrate TensorBoard with our model.
Running TensorBoard inside a notebook
First, we need to load Tensorboard’s extension for Jupyter:
Then, we run Tensorboard using the newly available magic:
The magic above tells TensorBoard to look for logs inside the folder specified by the logdir argument: runs. So, ...