We Raised $8M Series A to Continue Building Experiment Tracking and Model Registry That “Just Works”

Read more

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

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

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

Couldn’t find the use case you were looking for?

Just get in touch and our ML team will
create a custom demo for you