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 supported10 Subscribers
Add a CodeTriage badge to pytorch-lightning
Help out
- Issues
- Fix LR scheduler behaviour with AMP
- Introduce training and eval modes for the strategy
- WandbLogger disables cloud checkpointing in Trainer default_root_dir
- torch.randn() + DDP + GANs are easy to get wrong with lightning
- Support non-conventional optimizers
- Integrating pytorch XLA when using multiple GPUs
- `num_training_batches` is `inf` in `configure_optimizers`
- Stochastic weight averaging without the learning rate scheduler
- Using overfit_batches does not work with BatchSampler as Sampler
- ModelCheckpoint tries to find monitored key on non-zero ranks
- Docs
- Python not yet supported