Skip to main content

Running PyTorch Lightning

Running PyTorch Lightning scripts and hyper parameter sweeps in Grid is easy using CLI or the Web UI. Grid and Lightning are optimized to work together!

Both CPU and GPU instances are supported with all necessary system drivers pre-installed and set up, so you can take advantage of multiprocessing and compute acceleration hardware with no additional configuration.

Example

In the Web UI Dashboard view, click on Hello CIFAR-10 demo, this invokes the New Run dialog pre-populated with this script: https://github.com/gridai/hello_mnists/blob/master/pl_cifar10.py

Choose the machines you want to run the program on, the hyper parameter strategy, optionally any script arguments such as learning rate specified as numpy distribution: --learning_rate "uniform(1e-5, 1e-3, 8)" and Start Run

As the experiments start running, view the metrics and artifacts as well as invoke Tensorboard in the Runs view

If using CLI is your preference, you can follow the CLI tutorial