PyTorch Lightning
PyTorch Lightning is a lightweight PyTorch wrapper for high-performance AI research. Use the Neptune integration to:
- Monitor the training live.
- Log hyperparameters, model checkpoint paths, metrics, and more.
- Visualize the logged metadata in the Neptune app.
Note that this integration isn't yet included in the official PyTorch Lightning release and it's built on top of pytorch-lightning==2.5.1rc2
.
The README
file contains complete instructions on installation and usage.