pytorch-lightning
https://github.com/pytorchlightning/pytorch-lightning
Python
The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported12 Subscribers
Add a CodeTriage badge to pytorch-lightning
Help out
- Issues
- Fix sync_dist epoch metrics inflated by floored integer batch-size accumulator
- feat(FSDP): enable FSDP strategy on CPU and add trainable checkpointing tests
- Preserve leafless hyperparameter values in `_flatten_dict`
- Fix FSDP device mesh initialization on CPU
- Fix FSDP mixed precision autocast on CPU
- Add an end-to-end anomaly detection tutorial using PyTorch Lightning
- feat: add `dtype` option to `LightningModule.load_from_checkpoint`
- Handle jagged NestedTensor failures in model summary
- feat: Log key prefix controls for Trainer, WandbLogger, and CometLogger
- Fix stateful dataloader state not restored on resume after `estimated_stepping_batches`
- Docs
- Python not yet supported