How to Track ML Model Training: PyTorch Lightning + Neptune Integration

1 min
Parth Tiwary
4th April, 2022

What will you get with this integration?

PyTorch Lightning is a lightweight PyTorch wrapper for high-performance AI research.

With Neptune integration you can automatically:

  • Monitor model training live
  • Log training, validation, and testing metrics, and visualize them in the Neptune UI
  • Log hyperparameters
  • Monitor hardware usage
  • Log any additional metrics
  • Log performance charts and images
  • Save model checkpoints
  • Do whatever you would expect from a modern ML metadata store

Read the documentation

Want to make your project easy to share and collaborate on?

    Contact with us

    This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.
    * - required fields