pytorch/torch/optim
Sergii Dymchenko 313557a613 Add missing import (#72840)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72840

Reviewed By: H-Huang

Differential Revision: D34242612

Pulled By: albanD

fbshipit-source-id: 3dd34de96dbf1ae8f3c3ea45888d211d95862c49
(cherry picked from commit d2650ffa75)
2022-02-15 19:43:54 +00:00
..
_multi_tensor Optim foreach cleanup for AdamW (#70484) 2022-02-15 18:02:08 +00:00
__init__.py
__init__.pyi
_functional.py Optim foreach cleanup for AdamW (#70484) 2022-02-15 18:02:08 +00:00
adadelta.py Optim foreach cleanup for Adadelta (#69980) 2022-02-09 16:52:12 +00:00
adadelta.pyi
adagrad.py Optim foreach cleanup for Adagrad (#69981) 2022-02-09 16:52:12 +00:00
adagrad.pyi
adam.py Optim foreach cleanup for Adam (#70295) 2022-02-15 18:02:08 +00:00
adam.pyi
adamax.py Optim foreach cleanup for Adamax (#69982) 2022-02-09 16:52:13 +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
lbfgs.pyi
lr_scheduler.py SequentialLR update _last_lr on step (#70558) 2022-01-07 10:36:35 -08: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
optimizer.py Integrate multi_tensor zero_grad into Optimizer base class (#69936) 2022-01-05 15:46:23 -08:00
optimizer.pyi
radam.py Optim foreach cleanup for RAdam (#70230) 2022-02-09 16:52:13 +00:00
radam.pyi
rmsprop.py Optim foreach cleanup for Rmsprop (#70482) 2022-02-15 18:02:08 +00:00
rmsprop.pyi
rprop.py Optim foreach cleanup for Rprop (#70483) 2022-02-15 18:02:08 +00:00
rprop.pyi
sgd.py Optim foreach cleanup for SGD (#70481) 2022-02-15 18:02:08 +00:00
sgd.pyi
sparse_adam.py
sparse_adam.pyi
swa_utils.py Add missing import (#72840) 2022-02-15 19:43:54 +00:00
swa_utils.pyi