Skip to content

Commit 97bfab4

Browse files
committed
refactor: load_optimizers
1 parent 49f3602 commit 97bfab4

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

pytorch_optimizer/optimizers.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,12 +34,10 @@ def load_optimizers(optimizer: str, use_fp16: bool = False):
3434
opt = AdaBound
3535
elif optimizer == 'madgrad':
3636
opt = MADGRAD
37-
elif optimizer == 'diffrgrad':
38-
opt = DiffRGrad
39-
elif optimizer == 'diffgrad':
40-
opt = DiffGrad
4137
elif optimizer == 'diffgrad':
4238
opt = DiffGrad
39+
elif optimizer == 'diffrgrad':
40+
opt = DiffRGrad
4341
elif optimizer == 'adahessian':
4442
opt = AdaHessian
4543
elif optimizer == 'lamb':

0 commit comments

Comments
 (0)