Taras Tsugrii 1c7ba330b2 [BE][optim] Simplify _init_group. (#120055) ... This version is more concise and avoids second lookup in case `momentum_buffer` is in the `state`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/120055 Approved by: https://github.com/janeyx99		2024-02-22 22:15:01 +00:00
..
_multi_tensor
__init__.py	[BE]: Enable F821 and fix bugs (#116579)	2024-01-01 08:40:46 +00:00
__init__.pyi
_functional.py
adadelta.py	Migrate test_complex_optimizer to OptimizerInfo (#118160)	2024-01-24 21:22:47 +00:00
adadelta.pyi
adagrad.py	Migrate test_complex_optimizer to OptimizerInfo (#118160)	2024-01-24 21:22:47 +00:00
adagrad.pyi
adam.py	[optim] Rectify capturable testing and fix bugs! (#118326)	2024-02-02 19:13:00 +00:00
adam.pyi
adamax.py	[optim] Rectify capturable testing and fix bugs! (#118326)	2024-02-02 19:13:00 +00:00
adamax.pyi
adamw.py	[optim] Rectify capturable testing and fix bugs! (#118326)	2024-02-02 19:13:00 +00:00
adamw.pyi
asgd.py	[optim] Rectify capturable testing and fix bugs! (#118326)	2024-02-02 19:13:00 +00:00
asgd.pyi
lbfgs.py	[optim] lbfgs: handle complex params as independent real params (#118184)	2024-01-31 19:24:16 +00:00
lbfgs.pyi
lr_scheduler.py	Clarify the patience in ReduceLROnPlateau (#119872)	2024-02-15 19:43:06 +00:00
lr_scheduler.pyi
nadam.py	[optim] Rectify capturable testing and fix bugs! (#118326)	2024-02-02 19:13:00 +00:00
nadam.pyi
optimizer.py	Revert "Remove extra graph breaks (#118987)"	2024-02-05 22:19:37 +00:00
radam.py	[optim] Rectify capturable testing and fix bugs! (#118326)	2024-02-02 19:13:00 +00:00
radam.pyi
rmsprop.py	Migrate test_complex_optimizer to OptimizerInfo (#118160)	2024-01-24 21:22:47 +00:00
rmsprop.pyi
rprop.py	[Optim][Rprop] Replace new().resize_as_() by torch.full_like() (#119978)	2024-02-16 19:54:04 +00:00
rprop.pyi
sgd.py	[BE][optim] Simplify _init_group. (#120055)	2024-02-22 22:15:01 +00:00
sgd.pyi
sparse_adam.py	Add guardrails preventing complex params in LBFGS & SparseAdam (#118161)	2024-01-24 21:22:47 +00:00
sparse_adam.pyi
swa_utils.py
swa_utils.pyi