OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
b2eb0e8c6a
pytorch/torch/optim
History
Wil Kong de06345e9b Avoid Host & Device Sync In LR Scheduler (#133663) 
Fixes #133662.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/133663
Approved by: https://github.com/janeyx99, https://github.com/eqy

Co-authored-by: Jane (Yuan) Xu <31798555+janeyx99@users.noreply.github.com>
2024-08-22 03:52:43 +00:00
..
_multi_tensor Optim package docstring fix (#129086) 2024-06-21 14:30:53 +00:00
__init__.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_adafactor.py Add Adafactor foreach impl (#132336) 2024-08-15 17:00:33 +00:00
_functional.py Flip default value for mypy disallow_untyped_defs [8/11] (#127845) 2024-06-08 18:49:56 +00:00
adadelta.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
adagrad.py move param's device check to _init_group for fused (#131153) 2024-08-17 04:49:47 +00:00
adam.py move param's device check to _init_group for fused (#131153) 2024-08-17 04:49:47 +00:00
adamax.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
adamw.py move param's device check to _init_group for fused (#131153) 2024-08-17 04:49:47 +00:00
asgd.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
lbfgs.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
lr_scheduler.py Avoid Host & Device Sync In LR Scheduler (#133663) 2024-08-22 03:52:43 +00:00
nadam.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
optimizer.py move param's device check to _init_group for fused (#131153) 2024-08-17 04:49:47 +00:00
radam.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
rmsprop.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
rprop.py Correct return type of grouping helper function in Optimizer (#133360) 2024-08-14 01:56:02 +00:00
sgd.py move param's device check to _init_group for fused (#131153) 2024-08-17 04:49:47 +00:00
sparse_adam.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
swa_utils.py Make optim.swa.util content accessible from the torch.optim doc (#133393) 2024-08-21 00:43:46 +00:00