Mikayla Gawarecki d9acfef831 Optim foreach cleanup for Adamax (#69982) ...

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/69982

Test Plan: Imported from OSS

Reviewed By: anjali411

Differential Revision: D33767865

Pulled By: mikaylagawarecki

fbshipit-source-id: c5efd351e359825d38b71f57a2c61a2055c3c114
(cherry picked from commit 37bb80c2d7b441c5718cee6f1b37653d4937e20a)

2022-02-09 16:52:13 +00:00

..

__init__.py

Optim foreach cleanup for Adamax (#69982)

2022-02-09 16:52:13 +00:00

__init__.pyi

Optim foreach cleanup for Adamax (#69982)

2022-02-09 16:52:13 +00:00

_functional.py

Optim foreach cleanup for Adamax (#69982)

2022-02-09 16:52:13 +00:00

adam.py

[optim] update step in functional and pass state_steps instead of state (#71333)

2022-02-08 16:51:19 +00:00

adam.pyi

[RELAND] Added optimizers based on multi tensor apply (#45408)

2020-09-28 13:14:04 -07:00

adamw.py

[optim] update step in functional and pass state_steps instead of state (#71333)

2022-02-08 16:51:19 +00:00

adamw.pyi

[RELAND] Added optimizers based on multi tensor apply (#45408)

2020-09-28 13:14:04 -07:00

asgd.py

[optim] ASGD fold state updates into functional and pass list of vars rather than states (#71335)

2022-02-08 23:58:41 +00:00

asgd.pyi

[RELAND] Added optimizers based on multi tensor apply (#45408)

2020-09-28 13:14:04 -07:00

nadam.py

[optim] NAdam fold state updates into functional (#71334)

2022-02-08 23:58:41 +00:00

nadam.pyi

To add Nesterov Adam algorithm for multi-tensor optimizers API (#59165)

2021-06-27 17:00:41 -07:00

radam.py

[optim] update step in functional and pass state_steps instead of state (#71333)

2022-02-08 16:51:19 +00:00

radam.pyi

To add Rectified Adam algorithm for multi-tensor optimizers API (#59161)

2021-06-27 13:01:20 -07:00

rmsprop.py

Integrate multi_tensor zero_grad into Optimizer base class (#69936)

2022-01-05 15:46:23 -08:00

rmsprop.pyi

[RELAND] Added optimizers based on multi tensor apply (#45408)

2020-09-28 13:14:04 -07:00

rprop.py

Integrate multi_tensor zero_grad into Optimizer base class (#69936)

2022-01-05 15:46:23 -08:00

rprop.pyi

[RELAND] Added optimizers based on multi tensor apply (#45408)

2020-09-28 13:14:04 -07:00

sgd.py

fix loading of older models that don't have maximize (#71023)

2022-01-10 06:01:24 -08:00

sgd.pyi

[RELAND] Added optimizers based on multi tensor apply (#45408)

2020-09-28 13:14:04 -07:00