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
- Add log_key_prefix to WandbLogger to control the prefix for global_step
- [quick video overview] Add epoch_key to CometLogger to make epoch extraction configurable
- Handle jagged NestedTensor failures in model summary
- 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