Adam doesn't work with nonzero-dim Tensor betas #147921
Labels
module: optimizer
Related to torch.optim
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
🐛 Describe the bug
This bug was pointed out at #145461 (comment). The PR #145674 fixed the Tensor
lr
issue, but not the Tensorbetas
issue.Versions
The same as #145461
cc @vincentqb @jbschlosser @albanD @janeyx99 @crcrpar
The text was updated successfully, but these errors were encountered: