lightning
https://github.com/lightning-ai/lightning
Python
Deep learning framework to train, deploy, and ship AI products Lightning fast.
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 supported5 Subscribers
Add a CodeTriage badge to lightning
Help out
- Issues
- [quick video overview] Add epoch_key to CometLogger to make epoch extraction configurable
- Handle jagged NestedTensor failures in model summary
- Add log_key_prefix to Trainer to control the prefix for metrics like epoch
- drop deprecated `allow_frozen` argument from `apply_to_collection` call
- feat: Log key prefix controls for Trainer, WandbLogger, and CometLogger
- fix: update version to 2.6.5
- fix(loops): increment batch_progress.completed before on_train_batch_end hooks
- Allow epochs with different size
- Fix: Skip redundant per-step device transfer of `_ResultMetric` in the logging hot path
- fix: preserve stateful dataloader state around estimated_stepping_batches
- Docs
- Python not yet supported