pytorch/torch/distributed/optim
2025-01-21 04:23:29 +00:00
..
__init__.py
_deprecation_warning.py
apply_optimizer_in_backward.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_adadelta.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_adagrad.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_adam.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_adamax.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_adamw.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_rmsprop.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_rprop.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
functional_sgd.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
named_optimizer.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
optimizer.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
post_localSGD_optimizer.py
utils.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
zero_redundancy_optimizer.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
zero_redundancy_optimizer.pyi