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