Michael Lazos 1fd4757fdc Support tensor betas in Adam and AdamW (#134171) ... Adds support for beta1 and beta2 to be wrapped in tensor for Adam and AdamW. Fixes https://github.com/pytorch/pytorch/issues/133898 Pull Request resolved: https://github.com/pytorch/pytorch/pull/134171 Approved by: https://github.com/janeyx99		2024-11-15 21:55:55 +00:00
..
_multi_tensor	Optim package docstring fix (#129086)	2024-06-21 14:30:53 +00:00
__init__.py	[BE][optim] Make pyright recognize exported symbols (#135043)	2024-09-04 21:53:46 +00:00
_adafactor.py	Add ScalarList overload to _foreach_lerp (#134482)	2024-11-12 19:03:41 +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-11-05 23:10:38 +00:00
adagrad.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-11-05 23:10:38 +00:00
adam.py	Support tensor betas in Adam and AdamW (#134171)	2024-11-15 21:55:55 +00:00
adamax.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-11-05 23:10:38 +00:00
adamw.py	Support tensor betas in Adam and AdamW (#134171)	2024-11-15 21:55:55 +00:00
asgd.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-11-05 23:10:38 +00:00
lbfgs.py	[BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771)	2024-08-01 17:07:14 +00:00
lr_scheduler.py	Remove unused Python variables in torch/[b-z]* (#136963)	2024-10-19 16:45:22 +00:00
nadam.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-11-05 23:10:38 +00:00
optimizer.py	Support tensor betas in Adam and AdamW (#134171)	2024-11-15 21:55:55 +00:00
radam.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-11-05 23:10:38 +00:00
rmsprop.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-11-05 23:10:38 +00:00
rprop.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-11-05 23:10:38 +00:00
sgd.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-11-05 23:10:38 +00:00
sparse_adam.py	Add Support for Tracking Parameter Names (named_parameters) in Optimizer State Dict (#134107)	2024-10-14 19:24:44 +00:00
swa_utils.py	Ensure SWA boundary conditions w.r.t. definition (#133773)	2024-10-31 18:24:08 +00:00