pytorch/torch/optim/_multi_tensor
ramvenkat98 4a544df00d Implement and benchmark a torch.optim.multi_tensor.adagrad implementation (#59155)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/59155

Test Plan: Imported from OSS

Reviewed By: iramazanli

Differential Revision: D29525213

Pulled By: ramvenkat98

fbshipit-source-id: 6d7e8da91c965d1f4e955a084ed875bab641dc9a
2021-07-07 08:08:32 -07:00
..
__init__.py Implement and benchmark a torch.optim.multi_tensor.adagrad implementation (#59155) 2021-07-07 08:08:32 -07:00
__init__.pyi Implement and benchmark a torch.optim.multi_tensor.adagrad implementation (#59155) 2021-07-07 08:08:32 -07:00
_functional.py Implement and benchmark a torch.optim.multi_tensor.adagrad implementation (#59155) 2021-07-07 08:08:32 -07:00
adadelta.py
adadelta.pyi
adagrad.py Implement and benchmark a torch.optim.multi_tensor.adagrad implementation (#59155) 2021-07-07 08:08:32 -07:00
adagrad.pyi Implement and benchmark a torch.optim.multi_tensor.adagrad implementation (#59155) 2021-07-07 08:08:32 -07:00
adam.py
adam.pyi
adamax.py
adamax.pyi
adamw.py
adamw.pyi
asgd.py
asgd.pyi
nadam.py To add Nesterov Adam algorithm for multi-tensor optimizers API (#59165) 2021-06-27 17:00:41 -07:00
nadam.pyi To add Nesterov Adam algorithm for multi-tensor optimizers API (#59165) 2021-06-27 17:00:41 -07:00
radam.py
radam.pyi
rmsprop.py
rmsprop.pyi
rprop.py
rprop.pyi
sgd.py
sgd.pyi