OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
0bd5a3fed7
pytorch/torch/optim
History
Tej Singh eafeba71c1 Adamw refactor (#115983) 
Fixes #104899, refactors adamw by abstracting out common code in adam.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/115983
Approved by: https://github.com/janeyx99
2023-12-17 06:58:39 +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 Adamw refactor (#115983) 2023-12-17 06:58:39 +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 Adamw refactor (#115983) 2023-12-17 06:58:39 +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 [torch] add use_buffers to swa_utils interface (#109078) 2023-09-19 21:30:59 +00:00