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 in the examples project.


The test loss and validation loss plots appear in the experiment's page in the ClearML web UI under 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 scalars


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

PyTorch Lightning parameters


Models created by the experiment appear in the experiment's ARTIFACTS tab.

PyTorch Lightning model

Clicking on a model name takes you to the model's page, where you can view the model's details and access the model.


All other console output appears in CONSOLE.

PyTorch Lightning console