pytorch
11047be1 - feat(optim): Add `NAdam`support for complex, with `has_complex` shortcut (#110634)

Commit
1 year ago
feat(optim): Add `NAdam`support for complex, with `has_complex` shortcut (#110634) Partial fix: https://github.com/pytorch/pytorch/issues/110606 More on `has_complex` shortcut: https://github.com/pytorch/pytorch/pull/110613#issuecomment-1749314805 CC: @janeyx99 @mlazos @lezcano Pull Request resolved: https://github.com/pytorch/pytorch/pull/110634 Approved by: https://github.com/lezcano
Author
Committer
Parents
Loading