pytorch
a41f00ed - [optim][sgd] group tensors in foreach to maximize perf (#92338)

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
Committer
Parents
Loading