The pytorch_tensorboard.py example demonstrates the integration of ClearML into code that uses PyTorch and TensorBoard.
The example does the following:
- Trains a simple deep neural network on the PyTorch built-in MNIST dataset.
- Creates a TensorBoard
SummaryWriterobject to log:
- Scalars during training.
- Scalars and debug samples during testing.
- Test text message to the console (a test message to demonstrate ClearML's automatic logging).
- Creates an experiment named
pytorch with tensorboard, which is associated with the
In the example script, the
test functions call the TensorBoard
SummaryWriter.add_scalar method to log loss.
These scalars, along with the resource utilization plots, which are titled :monitor: machine, appear in the experiment's page in the ClearML web UI under RESULTS > SCALARS,
ClearML automatically tracks images and text output to TensorFlow. They appear in RESULTS > DEBUG SAMPLES.
ClearML automatically logs TensorFlow Definitions. They appear in CONFIGURATIONS > HYPER PARAMETERS > TF_DEFINE.
Text printed to the console for training progress, as well as all other console output, appear in RESULTS > CONSOLE.
Model artifacts associated with the experiment appear in the info panel of the EXPERIMENTS tab and in the info panel of the MODELS tab.
The experiment info panel shows model tracking, including the model name and design (in this case, no design was stored).
The model info panel contains the model details, including:
- Model URL
- Snapshot locations.