pytorch/torch/optim
Rob Zinkov 2a496e2f80 Adding maximize to Adamax (#77409)
Added the maximize flag #68052 to Adamax optimizer and updates the respective tests.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77409
Approved by: https://github.com/albanD
2022-05-16 17:34:44 +00:00
..
_multi_tensor
__init__.py
__init__.pyi
_functional.py
adadelta.py
adadelta.pyi
adagrad.py Adding maximize flag to Adagrad 2022-04-20 08:29:03 +00:00
adagrad.pyi
adam.py
adam.pyi
adamax.py Adding maximize to Adamax (#77409) 2022-05-16 17:34:44 +00:00
adamax.pyi
adamw.py
adamw.pyi
asgd.py
asgd.pyi
lbfgs.py
lbfgs.pyi
lr_scheduler.py Fix CosineAnnealingLR on restart 2022-04-20 13:35:01 +00:00
lr_scheduler.pyi
nadam.py
nadam.pyi
optimizer.py
optimizer.pyi Fix Optimizer.zero_grad type annotation (#76998) 2022-05-11 00:05:26 +00:00
radam.py
radam.pyi
rmsprop.py
rmsprop.pyi
rprop.py Docs: Add missing zero-ing step in Rprop algorithm 2022-04-11 21:57:13 +00:00
rprop.pyi
sgd.py DOC fix momentum equation for nesterov 2022-05-04 20:40:21 +00:00
sgd.pyi
sparse_adam.py
sparse_adam.pyi
swa_utils.py
swa_utils.pyi