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 supported7 Subscribers
Add a CodeTriage badge to pytorch-lightning
Help out
- Issues
- Validation runs only for one iteration when restarting from checkpoint mid-epoch, wrongly reporting validation loss
- Add Ascend NPU as a backend
- DDP training timeout
- FSDP hybrid shard should checkpoint in a single node
- Gather tensors of unequal shapes
- "backward pass is invalid for module in evaluation mode" with deepspeed stage 3
- FSDP checkpointing uses deprecated APIs with PyTorch 2.2
- `batch_sampler.batch_size` is None with deepspeed and `DataLoader(batch_size=None)`
- How to log artifacts on rank > 0?
- Logging failed when using mlflow with sqlite tracking uri
- Docs
- Python not yet supported