pytorch
https://github.com/pytorch/pytorch
Python
Tensors and Dynamic neural networks in Python with strong GPU acceleration
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 supported916 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- [feature request] [dataloader] Pad variable-sized tensors in default_collate
- docs for torch.cuda.reset_max_memory_reserved don't exist
- torch.cuda.default_generators documentation referenced but don't exist.
- Scripting torchvision.models.detection.maskrcnn_resnet50_fpn
- [FR][RFC] Build a serving framework to host and serve trained PyTorch models
- Improve the error message when trying to install in a 32-bit Python environment
- Deployment training model at C + + end
- Version number is still duplicated in a bunch of places
- Avoid RTTI in DistEngine
- I can't set gpu is 1 it always use gpu 0
- Docs
- Python not yet supported