pytorch/torch/optim
2024-07-09 18:58:07 +00:00
..
_multi_tensor Optim package docstring fix (#129086) 2024-06-21 14:30:53 +00:00
__init__.py Remove circular import (#128875) 2024-06-18 12:30:13 +00:00
_functional.py Flip default value for mypy disallow_untyped_defs [8/11] (#127845) 2024-06-08 18:49:56 +00:00
adadelta.py Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-06-10 20:44:42 +00:00
adagrad.py Flip default value for mypy disallow_untyped_defs [8/11] (#127845) 2024-06-08 18:49:56 +00:00
adam.py [MPS] Add tensor_lr overloads to fused adam & adamw (#129451) 2024-07-02 19:46:30 +00:00
adamax.py Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-06-10 20:44:42 +00:00
adamw.py [MPS] Add tensor_lr overloads to fused adam & adamw (#129451) 2024-07-02 19:46:30 +00:00
asgd.py Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-06-10 20:44:42 +00:00
lbfgs.py Flip default value for mypy disallow_untyped_defs [8/11] (#127845) 2024-06-08 18:49:56 +00:00
lr_scheduler.py fix the use of initial learning rate in the OneCycleLR example (#130306) 2024-07-09 18:58:07 +00:00
nadam.py Optim package docstring fix (#129086) 2024-06-21 14:30:53 +00:00
optimizer.py Optim package docstring fix (#129086) 2024-06-21 14:30:53 +00:00
radam.py Optim package docstring fix (#129086) 2024-06-21 14:30:53 +00:00
rmsprop.py Optim package docstring fix (#129086) 2024-06-21 14:30:53 +00:00
rprop.py Optim package docstring fix (#129086) 2024-06-21 14:30:53 +00:00
sgd.py Optim package docstring fix (#129086) 2024-06-21 14:30:53 +00:00
sparse_adam.py Flip default value for mypy disallow_untyped_defs [8/11] (#127845) 2024-06-08 18:49:56 +00:00
swa_utils.py Optim package docstring fix (#129086) 2024-06-21 14:30:53 +00:00