pytorch
0157e2ef - [optim][adamw] default to foreach when CUDA + differentiable=False (#92306)

Commit
2 years ago
[optim][adamw] default to foreach when CUDA + differentiable=False (#92306) Pull Request resolved: https://github.com/pytorch/pytorch/pull/92306 Approved by: https://github.com/albanD
Author
Committer
Parents
Loading