pytorch
30bda6b1 - [optim][adamax] default to foreach when CUDA + differentiable=False (#92723)

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