pytorch/torch/optim
PyTorch MergeBot 5b6b680517 Revert "Adamw refactor (#115983)"
This reverts commit eafeba71c1.

Reverted https://github.com/pytorch/pytorch/pull/115983 on behalf of https://github.com/jeanschmidt due to Breaking internal tests, @janeyx99 please help @tfsingh to have this PR landed ([comment](https://github.com/pytorch/pytorch/pull/115983#issuecomment-1862976954))
2023-12-19 15:26:44 +00:00
..
_multi_tensor
__init__.py Fixed docstring errors inside torch/cuda/ and torch/optim/ (Docathon H2) (#112964) 2023-11-13 22:16:44 +00:00
__init__.pyi
_functional.py Fixed docstring errors inside torch/cuda/ and torch/optim/ (Docathon H2) (#112964) 2023-11-13 22:16:44 +00:00
adadelta.py Added More Information About Adadelta Optimizer (#106290) 2023-12-05 05:55:16 +00:00
adadelta.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
adagrad.py [optim] be explicit about CPU scalar tensor dtypes (#111008) 2023-11-21 22:44:50 +00:00
adagrad.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
adam.py Revert "Adamw refactor (#115983)" 2023-12-19 15:26:44 +00:00
adam.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
adamax.py [optim] be explicit about CPU scalar tensor dtypes (#111008) 2023-11-21 22:44:50 +00:00
adamax.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
adamw.py Revert "Adamw refactor (#115983)" 2023-12-19 15:26:44 +00:00
adamw.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
asgd.py [optim] be explicit about CPU scalar tensor dtypes (#111008) 2023-11-21 22:44:50 +00:00
asgd.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
lbfgs.py Fixed docstring errors inside torch/cuda/ and torch/optim/ (Docathon H2) (#112964) 2023-11-13 22:16:44 +00:00
lbfgs.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
lr_scheduler.py Solving pickle error when saving CyclicLR state_dict (#110931) 2023-11-22 11:38:35 +00:00
lr_scheduler.pyi fixed pyi file for ReduceLROnPlateau (#113659) 2023-11-15 19:33:36 +00:00
nadam.py [optim] be explicit about CPU scalar tensor dtypes (#111008) 2023-11-21 22:44:50 +00:00
nadam.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
optimizer.py [BE][SparseAdam] cleaner way to verify no sparse params (#114425) 2023-11-29 19:47:03 +00:00
radam.py [optim] be explicit about CPU scalar tensor dtypes (#111008) 2023-11-21 22:44:50 +00:00
radam.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
rmsprop.py feat(optim): use has_complex shortcut flag for all applicable optimizers, use _view_as_real auxiliary function (#110706) 2023-10-31 20:33:03 +00:00
rmsprop.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
rprop.py feat(optim): use has_complex shortcut flag for all applicable optimizers, use _view_as_real auxiliary function (#110706) 2023-10-31 20:33:03 +00:00
rprop.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
sgd.py Set default LR value of SGD to 1e-3 (#114467) 2023-11-23 19:07:38 +00:00
sgd.pyi Set default LR value of SGD to 1e-3 (#114467) 2023-11-23 19:07:38 +00:00
sparse_adam.py [BE][SparseAdam] cleaner way to verify no sparse params (#114425) 2023-11-29 19:47:03 +00:00
sparse_adam.pyi [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
swa_utils.py
swa_utils.pyi