pytorch
6ba041fc - Look up `group["capturable"]`, not `defaults["capturable"]` in Adam(W) (#94149)

Commit
2 years ago
Look up `group["capturable"]`, not `defaults["capturable"]` in Adam(W) (#94149) We could set different values in each `param_group` when calling dunder init of `torch.optim` optimizers as in e.g. https://github.com/pytorch/pytorch/issues/89987. So check whether or not `capturable` is `True` among all the `param_group`s. Pull Request resolved: https://github.com/pytorch/pytorch/pull/94149 Approved by: https://github.com/albanD
Author
Committer
Parents
Loading