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 [PyTorch/d2go] fix optim _multi_tensor (#73215) 2022-02-23 10:29:48 +00:00
__init__.py To add Rectified Adam Algorithm to Optimizers (#58968) 2021-06-23 18:27:57 -07:00
__init__.pyi To add Rectified Adam Algorithm to Optimizers (#58968) 2021-06-23 18:27:57 -07:00
_functional.py Optim foreach cleanup for AdamW (#70484) 2022-02-15 18:02:08 +00:00
adadelta.py Add maximize flag to Adadelta 2022-04-08 20:32:35 +00:00
adadelta.pyi
adagrad.py Adding maximize flag to Adagrad 2022-04-20 08:29:03 +00:00
adagrad.pyi
adam.py Optim foreach cleanup for Adam (#70295) 2022-02-15 18:02:08 +00:00
adam.pyi
adamax.py Adding maximize to Adamax (#77409) 2022-05-16 17:34:44 +00:00
adamax.pyi
adamw.py Optim foreach cleanup for AdamW (#70484) 2022-02-15 18:02:08 +00:00
adamw.pyi
asgd.py Optim foreach cleanup for ASGD (#70231) 2022-02-09 16:52:13 +00:00
asgd.pyi
lbfgs.py breakup optim, cuda documentation (#55673) 2021-04-14 12:44:00 -07:00
lbfgs.pyi
lr_scheduler.py Fix CosineAnnealingLR on restart 2022-04-20 13:35:01 +00:00
lr_scheduler.pyi update SequentialLR signature (#69817) 2021-12-16 12:58:00 -08:00
nadam.py Optim foreach cleanup for NAdam (#70229) 2022-02-09 16:52:13 +00:00
nadam.pyi To add Nesterov Adam Algorithm to Optimizers (#59009) 2021-06-23 08:21:43 -07:00
optimizer.py Fix casting bug in state_step for optimizers when loading state dict 2022-04-05 01:27:18 +00:00
optimizer.pyi Fix Optimizer.zero_grad type annotation (#76998) 2022-05-11 00:05:26 +00:00
radam.py Optim foreach cleanup for RAdam (#70230) 2022-02-09 16:52:13 +00:00
radam.pyi To add Rectified Adam Algorithm to Optimizers (#58968) 2021-06-23 18:27:57 -07:00
rmsprop.py Optim foreach cleanup for Rmsprop (#70482) 2022-02-15 18:02:08 +00:00
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 To refactor Sparse Adam algorithm for functional form (#59171) 2021-06-25 06:35:39 -07:00
sparse_adam.pyi
swa_utils.py Add missing import (#72840) 2022-02-15 19:43:54 +00:00
swa_utils.pyi Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00