mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
API to retrieve in backward optimizer for checkpointing purposes Differential Revision: [D47782225](https://our.internmc.facebook.com/intern/diff/D47782225/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/105991 Approved by: https://github.com/awgu |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| apply_optimizer_in_backward.py | ||
| functional_adadelta.py | ||
| functional_adagrad.py | ||
| functional_adam.py | ||
| functional_adamax.py | ||
| functional_adamw.py | ||
| functional_rmsprop.py | ||
| functional_rprop.py | ||
| functional_sgd.py | ||
| named_optimizer.py | ||
| optimizer.py | ||
| post_localSGD_optimizer.py | ||
| utils.py | ||
| zero_redundancy_optimizer.py | ||
| zero_redundancy_optimizer.pyi | ||