pytorch/torch/optim
Philip Meier 8423ab4f99 Fix CosineAnnealingWarmRestart annotation (#61106)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/44770.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/61106

Reviewed By: 1ntEgr8

Differential Revision: D29635764

Pulled By: walterddr

fbshipit-source-id: ddc45a7f04532a76d033ae7774706da1fa8608f7
2021-07-09 08:28:18 -07:00
..
_multi_tensor Implement and benchmark a torch.optim.multi_tensor.adagrad implementation (#59155) 2021-07-07 08:08:32 -07: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 To refactor Sparse Adam algorithm for functional form (#59171) 2021-06-25 06:35:39 -07:00
adadelta.py [optim] take kw-only argument for functional optim APIs (#56185) 2021-04-15 20:08:04 -07:00
adadelta.pyi
adagrad.py [optim] take kw-only argument for functional optim APIs (#56185) 2021-04-15 20:08:04 -07:00
adagrad.pyi
adam.py [optim] take kw-only argument for functional optim APIs (#56185) 2021-04-15 20:08:04 -07:00
adam.pyi
adamax.py [optim] take kw-only argument for functional optim APIs (#56185) 2021-04-15 20:08:04 -07:00
adamax.pyi
adamw.py [optim] take kw-only argument for functional optim APIs (#56185) 2021-04-15 20:08:04 -07:00
adamw.pyi
asgd.py refactor ASGD to use functional API (#58410) 2021-05-19 18:55:52 -07:00
asgd.pyi
lbfgs.py
lbfgs.pyi
lr_scheduler.py
lr_scheduler.pyi Fix CosineAnnealingWarmRestart annotation (#61106) 2021-07-09 08:28:18 -07:00
nadam.py To add Nesterov Adam Algorithm to Optimizers (#59009) 2021-06-23 08:21:43 -07:00
nadam.pyi To add Nesterov Adam Algorithm to Optimizers (#59009) 2021-06-23 08:21:43 -07:00
optimizer.py
optimizer.pyi
radam.py To add Rectified Adam Algorithm to Optimizers (#58968) 2021-06-23 18:27:57 -07:00
radam.pyi To add Rectified Adam Algorithm to Optimizers (#58968) 2021-06-23 18:27:57 -07:00
rmsprop.py [optim] take kw-only argument for functional optim APIs (#56185) 2021-04-15 20:08:04 -07:00
rmsprop.pyi
rprop.py [optim] take kw-only argument for functional optim APIs (#56185) 2021-04-15 20:08:04 -07:00
rprop.pyi
sgd.py [optim] take kw-only argument for functional optim APIs (#56185) 2021-04-15 20:08:04 -07: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
swa_utils.pyi