pytorch
30bda6b1
- [optim][adamax] default to foreach when CUDA + differentiable=False (#92723)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[optim][adamax] default to foreach when CUDA + differentiable=False (#92723) Pull Request resolved: https://github.com/pytorch/pytorch/pull/92723 Approved by: https://github.com/albanD
Author
janeyx99
Committer
pytorchmergebot
Parents
9b4a7784
Loading