pytorch
a41f00ed
- [optim][sgd] group tensors in foreach to maximize perf (#92338)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[optim][sgd] group tensors in foreach to maximize perf (#92338) Make foreach faster for SGD Pull Request resolved: https://github.com/pytorch/pytorch/pull/92338 Approved by: https://github.com/albanD
Author
janeyx99
Committer
pytorchmergebot
Parents
98b78aa1
Loading