File tree Expand file tree Collapse file tree 4 files changed +33
-0
lines changed Expand file tree Collapse file tree 4 files changed +33
-0
lines changed Original file line number Diff line number Diff line change @@ -61,6 +61,9 @@ Supported Optimizers
61
61
| `PID `_ | http://www4.comp.polyu.edu.hk/~cslzhang/paper/CVPR18_PID.pdf |
62
62
+-------------+-------------------------------------------------------------------------------+
63
63
| | |
64
+ | `QHAdam `_ | https://arxiv.org/abs/1810.06801 |
65
+ +-------------+-------------------------------------------------------------------------------+
66
+ | | |
64
67
| `QHM `_ | https://arxiv.org/abs/1810.06801 |
65
68
+-------------+-------------------------------------------------------------------------------+
66
69
| | |
307
310
**Reference Code **: https://github.com/tensorboy/PIDOptimizer
308
311
309
312
313
+ QHAdam
314
+ ------
315
+
316
+ +----------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------+
317
+ | .. image:: https://raw.githubusercontent.com/jettify/pytorch-optimizer/master/docs/rastrigin_QHAdam.png | .. image:: https://raw.githubusercontent.com/jettify/pytorch-optimizer/master/docs/rosenbrock_QHAdam.png |
318
+ +----------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------+
319
+
320
+ .. code :: python
321
+
322
+ import torch_optimizer as optim
323
+
324
+ # model = ...
325
+ optimizer = optim.QHAdam(
326
+ m.parameters(),
327
+ lr = 1e-3 ,
328
+ betas = (0.9 , 0.999 ),
329
+ nus = (1.0 , 1.0 ),
330
+ weight_decay = 0 ,
331
+ decouple_weight_decay = False ,
332
+ eps = 1e-8 ,
333
+ )
334
+ optimizer.step()
335
+
336
+
337
+ **Paper **: *Quasi-hyperbolic momentum and Adam for deep learning * (2019) [https://arxiv.org/abs/1810.06801]
338
+
339
+ **Reference Code **: https://github.com/facebookresearch/qhoptim
340
+
341
+
310
342
QHM
311
343
---
312
344
Original file line number Diff line number Diff line change @@ -171,6 +171,7 @@ def execute_experiments(
171
171
(optim .SGDW , - 8 , - 1.5 ),
172
172
(optim .PID , - 8 , - 1.0 ),
173
173
(optim .QHM , - 6 , - 0.2 ),
174
+ (optim .QHAdam , - 8 , 0.1 ),
174
175
]
175
176
execute_experiments (
176
177
optimizers ,
You can’t perform that action at this time.
0 commit comments