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 supported8 Subscribers
Add a CodeTriage badge to pytorch-lightning
Help out
- Issues
- environment variable WORLD_SIZE is incorrectly set to 1 after trainer.fit is done
- KeyError: 'Trying to restore optimizer state but checkpoint contains only the model. This is probably due to `ModelCheckpoint.save_weights_only` being set to `True`.' But optim_cfg is in model
- torch.cuda.OutOfMemoryError after running tuner.scale_batch_size() in "binsearch" mode
- Dashboard
- "FileExistsError: [Errno 17] File exists: '/000000_epoch_shape'" using the ddp_notebook strategy with data stored in MDS (mosaic streaming) format
- metric.compute() hangs when using DDP with multiple GPUs
- Can no longer install versions 1.5.10-1.6.5
- Switching into training mode in training_step
- Questions about loading a pre-trained model using lightnining CLI for continue training
- using deepspeed in pytorch lightning, a bug occurred : RuntimeError: Function ConvolutionBackward0 returned an invalid gradient at index 1
- Docs
- Python not yet supported