Skip to main content

PyTorch Lightning

The pytorch-lightning script demonstrates the integration of ClearML into code that uses PyTorch Lightning.

The example script does the following:

  • Trains a simple deep neural network on the PyTorch built-in MNIST dataset
  • Defines Argparse command line options, which are automatically captured by ClearML
  • Creates an experiment named pytorch lightning mnist example, which is associated with the examples project.

Scalars#

The test loss and validation loss plots appear in the experiment's page in the ClearML web UI under RESULTS > SCALARS. Resource utilization plots, which are titled :monitor: machine, also appear in the SCALARS tab. All of these plots are automatically captured by ClearML.

PyTorch Lightning console

Hyperparameters#

ClearML automatically logs command line options defined with argparse and TensorFlow Definitions, which appear in CONFIGURATIONS > HYPER PARAMETERS > Args and TF_DEFINE respectively.

PyTorch Lightning parameters

Artifacts#

Model artifacts associated with the experiment appear in the info panel of the EXPERIMENTS tab and in the info panel of the MODELS tab.

PyTorch Lightning model

Console#

All other console output appears in RESULTS > CONSOLE.

PyTorch Lightning console