pytorch-optimizer 2.11.0__tar.gz → 2.11.2__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/PKG-INFO +155 -146
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/README.rst +152 -143
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pyproject.toml +14 -11
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/__init__.py +12 -2
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/focal.py +20 -0
- pytorch_optimizer-2.11.2/pytorch_optimizer/loss/lovasz.py +50 -0
- pytorch_optimizer-2.11.2/pytorch_optimizer/loss/tversky.py +31 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adabelief.py +1 -1
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adafactor.py +1 -1
- pytorch_optimizer-2.11.2/pytorch_optimizer/optimizer/came.py +220 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/dadapt.py +129 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/fp16.py +1 -1
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/lomo.py +1 -1
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/lookahead.py +15 -3
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/rotograd.py +2 -2
- pytorch_optimizer-2.11.2/pytorch_optimizer/optimizer/tiger.py +90 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/LICENSE +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/base/__init__.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/base/exception.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/base/optimizer.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/base/scheduler.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/base/types.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/__init__.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/bi_tempered.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/cross_entropy.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/dice.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/f1.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/jaccard.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/loss/ldam.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/__init__.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/chebyshev.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/cosine_anealing.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/experimental/__init__.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/experimental/deberta_v3_lr_scheduler.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/linear_warmup.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/lr_scheduler/proportion.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/__init__.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/a2grad.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adabound.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adadelta.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adahessian.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adai.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adamax.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adamod.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adamp.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adams.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adan.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adanorm.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adapnm.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adashift.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/adasmooth.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/agc.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/aggmo.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/alig.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/amos.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/apollo.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/avagrad.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/diffgrad.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/fromage.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/gc.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/gravity.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/gsam.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/lamb.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/lars.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/lion.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/madgrad.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/msvag.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/nero.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/novograd.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/padam.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/pcgrad.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/pid.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/pnm.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/prodigy.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/qhadam.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/qhm.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/radam.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/ranger.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/ranger21.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/sam.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/sgd.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/sgdp.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/shampoo.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/shampoo_utils.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/sm3.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/sophia.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/srmm.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/swats.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/utils.py +0 -0
- {pytorch_optimizer-2.11.0 → pytorch_optimizer-2.11.2}/pytorch_optimizer/optimizer/yogi.py +0 -0
|
@@ -1,10 +1,10 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
|
-
Name:
|
|
3
|
-
Version: 2.11.
|
|
2
|
+
Name: pytorch_optimizer
|
|
3
|
+
Version: 2.11.2
|
|
4
4
|
Summary: optimizer & lr scheduler & objective function collections in PyTorch
|
|
5
5
|
Home-page: https://github.com/kozistr/pytorch_optimizer
|
|
6
6
|
License: Apache-2.0
|
|
7
|
-
Keywords: pytorch,deep-learning,optimizer,lr scheduler,A2Grad,ASGD,AccSGD,AdaBelief,AdaBound,AdaDelta,AdaFactor,AdaMax,AdaMod,AdaNorm,AdaPNM,AdaSmooth,AdaHessian,Adai,AdamP,AdamS,Adan,AggMo,AliG,Amos,Apollo,AvaGrad,DAdaptAdaGrad,DAdaptAdam,DAdaptAdan,DAdaptSGD,DiffGrad,Fromage,Gravity,GSAM,LARS,Lamb,Lion,LOMO,Lookahead,MADGRAD,MSVAG,Nero,NovoGrad,PAdam,PCGrad,PID,PNM,Prodigy,QHAdam,QHM,RAdam,Ranger,Ranger21,RotoGrad,SAM,SGDP,SGDW,SignSGD,SM3,SopihaH,SRMM,SWATS,
|
|
7
|
+
Keywords: pytorch,deep-learning,optimizer,lr scheduler,A2Grad,ASGD,AccSGD,AdaBelief,AdaBound,AdaDelta,AdaFactor,AdaMax,AdaMod,AdaNorm,AdaPNM,AdaSmooth,AdaHessian,Adai,AdamP,AdamS,Adan,AggMo,AliG,Amos,Apollo,AvaGrad,CAME,DAdaptAdaGrad,DAdaptAdam,DAdaptAdan,DAdaptSGD,DAdaptLion,DiffGrad,Fromage,Gravity,GSAM,LARS,Lamb,Lion,LOMO,Lookahead,MADGRAD,MSVAG,Nero,NovoGrad,PAdam,PCGrad,PID,PNM,Prodigy,QHAdam,QHM,RAdam,Ranger,Ranger21,RotoGrad,SAM,SGDP,Shampoo,ScalableShampoo,SGDW,SignSGD,SM3,SopihaH,SRMM,SWATS,Tiger,Yogi,BCE,BCEFocal,Focal,FocalCosine,SoftF1,Dice,LDAM,Jaccard,Bi-Tempered,Tversky,FocalTversky,LovaszHinge
|
|
8
8
|
Author: kozistr
|
|
9
9
|
Author-email: kozistr@gmail.com
|
|
10
10
|
Maintainer: kozistr
|
|
@@ -54,7 +54,7 @@ pytorch-optimizer
|
|
|
54
54
|
|
|
55
55
|
| **pytorch-optimizer** is optimizer & lr scheduler collections in PyTorch.
|
|
56
56
|
| I just re-implemented (speed & memory tweaks, plug-ins) the algorithm while based on the original paper. Also, It includes useful and practical optimization ideas.
|
|
57
|
-
| Currently,
|
|
57
|
+
| Currently, **60 optimizers**, **10 lr schedulers**, and **13 loss functions** are supported!
|
|
58
58
|
|
|
|
59
59
|
| Highly inspired by `pytorch-optimizer <https://github.com/jettify/pytorch-optimizer>`__.
|
|
60
60
|
|
|
@@ -69,20 +69,20 @@ So, please double-check the license before using it at your work.
|
|
|
69
69
|
Installation
|
|
70
70
|
~~~~~~~~~~~~
|
|
71
71
|
|
|
72
|
-
::
|
|
72
|
+
.. code-block:: bash
|
|
73
73
|
|
|
74
74
|
$ pip3 install -U pytorch-optimizer
|
|
75
75
|
|
|
76
76
|
If there's a version issue when installing the package, try with `--no-deps` option.
|
|
77
77
|
|
|
78
|
-
::
|
|
78
|
+
.. code-block:: bash
|
|
79
79
|
|
|
80
80
|
$ pip3 install -U --no-deps pytorch-optimizer
|
|
81
81
|
|
|
82
82
|
Simple Usage
|
|
83
83
|
~~~~~~~~~~~~
|
|
84
84
|
|
|
85
|
-
::
|
|
85
|
+
.. code-block:: python
|
|
86
86
|
|
|
87
87
|
from pytorch_optimizer import AdamP
|
|
88
88
|
|
|
@@ -99,7 +99,7 @@ Simple Usage
|
|
|
99
99
|
|
|
100
100
|
Also, you can load the optimizer via `torch.hub`
|
|
101
101
|
|
|
102
|
-
::
|
|
102
|
+
.. code-block:: python
|
|
103
103
|
|
|
104
104
|
import torch
|
|
105
105
|
|
|
@@ -109,7 +109,7 @@ Also, you can load the optimizer via `torch.hub`
|
|
|
109
109
|
|
|
110
110
|
If you want to build the optimizer with parameters & configs, there's `create_optimizer()` API.
|
|
111
111
|
|
|
112
|
-
::
|
|
112
|
+
.. code-block:: python
|
|
113
113
|
|
|
114
114
|
from pytorch_optimizer import create_optimizer
|
|
115
115
|
|
|
@@ -127,144 +127,148 @@ Supported Optimizers
|
|
|
127
127
|
|
|
128
128
|
You can check the supported optimizers with below code.
|
|
129
129
|
|
|
130
|
-
::
|
|
130
|
+
.. code-block:: python
|
|
131
131
|
|
|
132
132
|
from pytorch_optimizer import get_supported_optimizers
|
|
133
133
|
|
|
134
134
|
supported_optimizers = get_supported_optimizers()
|
|
135
135
|
|
|
136
|
-
|
|
137
|
-
| Optimizer | Description | Official Code
|
|
138
|
-
|
|
139
|
-
| AdaBelief | *Adapting Step-sizes by the Belief in Observed Gradients* | `github <https://github.com/juntang-zhuang/Adabelief-Optimizer>`__
|
|
140
|
-
|
|
141
|
-
| AdaBound | *Adaptive Gradient Methods with Dynamic Bound of Learning Rate* | `github <https://github.com/Luolc/AdaBound/blob/master/adabound/adabound.py>`__
|
|
142
|
-
|
|
143
|
-
| AdaHessian | *An Adaptive Second Order Optimizer for Machine Learning* | `github <https://github.com/amirgholami/adahessian>`__
|
|
144
|
-
|
|
145
|
-
| AdamD | *Improved bias-correction in Adam* |
|
|
146
|
-
|
|
147
|
-
| AdamP | *Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights* | `github <https://github.com/clovaai/AdamP>`__
|
|
148
|
-
|
|
149
|
-
| diffGrad | *An Optimization Method for Convolutional Neural Networks* | `github <https://github.com/shivram1987/diffGrad>`__
|
|
150
|
-
|
|
151
|
-
| MADGRAD | *A Momentumized, Adaptive, Dual Averaged Gradient Method for Stochastic* | `github <https://github.com/facebookresearch/madgrad>`__
|
|
152
|
-
|
|
153
|
-
| RAdam | *On the Variance of the Adaptive Learning Rate and Beyond* | `github <https://github.com/LiyuanLucasLiu/RAdam>`__
|
|
154
|
-
|
|
155
|
-
| Ranger | *a synergistic optimizer combining RAdam and LookAhead, and now GC in one optimizer* | `github <https://github.com/lessw2020/Ranger-Deep-Learning-Optimizer>`__
|
|
156
|
-
|
|
157
|
-
| Ranger21 | *a synergistic deep learning optimizer* | `github <https://github.com/lessw2020/Ranger21>`__
|
|
158
|
-
|
|
159
|
-
| Lamb | *Large Batch Optimization for Deep Learning* | `github <https://github.com/cybertronai/pytorch-lamb>`__
|
|
160
|
-
|
|
161
|
-
| Shampoo | *Preconditioned Stochastic Tensor Optimization* | `github <https://github.com/moskomule/shampoo.pytorch>`__
|
|
162
|
-
|
|
163
|
-
| Nero | *Learning by Turning: Neural Architecture Aware Optimisation* | `github <https://github.com/jxbz/nero>`__
|
|
164
|
-
|
|
165
|
-
| Adan | *Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models* | `github <https://github.com/sail-sg/Adan>`__
|
|
166
|
-
|
|
167
|
-
| Adai | *Disentangling the Effects of Adaptive Learning Rate and Momentum* | `github <https://github.com/zeke-xie/adaptive-inertia-adai>`__
|
|
168
|
-
|
|
169
|
-
| SAM | *Sharpness-Aware Minimization* | `github <https://github.com/davda54/sam>`__
|
|
170
|
-
|
|
171
|
-
| ASAM | *Adaptive Sharpness-Aware Minimization* | `github <https://github.com/davda54/sam>`__
|
|
172
|
-
|
|
173
|
-
| GSAM | *Surrogate Gap Guided Sharpness-Aware Minimization* | `github <https://github.com/juntang-zhuang/GSAM>`__
|
|
174
|
-
|
|
175
|
-
| D-Adaptation | *Learning-Rate-Free Learning by D-Adaptation* | `github <https://github.com/facebookresearch/dadaptation>`__
|
|
176
|
-
|
|
177
|
-
| AdaFactor | *Adaptive Learning Rates with Sublinear Memory Cost* | `github <https://github.com/DeadAt0m/adafactor-pytorch>`__
|
|
178
|
-
|
|
179
|
-
| Apollo | *An Adaptive Parameter-wise Diagonal Quasi-Newton Method for Nonconvex Stochastic Optimization* | `github <https://github.com/XuezheMax/apollo>`__
|
|
180
|
-
|
|
181
|
-
| NovoGrad | *Stochastic Gradient Methods with Layer-wise Adaptive Moments for Training of Deep Networks* | `github <https://github.com/lonePatient/NovoGrad-pytorch>`__
|
|
182
|
-
|
|
183
|
-
| Lion | *Symbolic Discovery of Optimization Algorithms* | `github <https://github.com/google/automl/tree/master/lion>`__
|
|
184
|
-
|
|
185
|
-
| Ali-G | *Adaptive Learning Rates for Interpolation with Gradients* | `github <https://github.com/oval-group/ali-g>`__
|
|
186
|
-
|
|
187
|
-
| SM3 | *Memory-Efficient Adaptive Optimization* | `github <https://github.com/google-research/google-research/tree/master/sm3>`__
|
|
188
|
-
|
|
189
|
-
| AdaNorm | *Adaptive Gradient Norm Correction based Optimizer for CNNs* | `github <https://github.com/shivram1987/AdaNorm>`__
|
|
190
|
-
|
|
191
|
-
| RotoGrad | *Gradient Homogenization in Multitask Learning* | `github <https://github.com/adrianjav/rotograd>`__
|
|
192
|
-
|
|
193
|
-
| A2Grad | *Optimal Adaptive and Accelerated Stochastic Gradient Descent* | `github <https://github.com/severilov/A2Grad_optimizer>`__
|
|
194
|
-
|
|
195
|
-
| AccSGD | *Accelerating Stochastic Gradient Descent For Least Squares Regression* | `github <https://github.com/rahulkidambi/AccSGD>`__
|
|
196
|
-
|
|
197
|
-
| SGDW | *Decoupled Weight Decay Regularization* | `github <https://github.com/loshchil/AdamW-and-SGDW>`__
|
|
198
|
-
|
|
199
|
-
| ASGD | *Adaptive Gradient Descent without Descent* | `github <https://github.com/ymalitsky/adaptive_GD>`__
|
|
200
|
-
|
|
201
|
-
| Yogi | *Adaptive Methods for Nonconvex Optimization* |
|
|
202
|
-
|
|
203
|
-
| SWATS | *Improving Generalization Performance by Switching from Adam to SGD* |
|
|
204
|
-
|
|
205
|
-
| Fromage | *On the distance between two neural networks and the stability of learning* | `github <https://github.com/jxbz/fromage>`__
|
|
206
|
-
|
|
207
|
-
| MSVAG | *Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradients* | `github <https://github.com/lballes/msvag>`__
|
|
208
|
-
|
|
209
|
-
| AdaMod | *An Adaptive and Momental Bound Method for Stochastic Learning* | `github <https://github.com/lancopku/AdaMod>`__
|
|
210
|
-
|
|
211
|
-
| AggMo | *Aggregated Momentum: Stability Through Passive Damping* | `github <https://github.com/AtheMathmo/AggMo>`__
|
|
212
|
-
|
|
213
|
-
| QHAdam | *Quasi-hyperbolic momentum and Adam for deep learning* | `github <https://github.com/facebookresearch/qhoptim>`__
|
|
214
|
-
|
|
215
|
-
| PID | *A PID Controller Approach for Stochastic Optimization of Deep Networks* | `github <https://github.com/tensorboy/PIDOptimizer>`__
|
|
216
|
-
|
|
217
|
-
| Gravity | *a Kinematic Approach on Optimization in Deep Learning* | `github <https://github.com/dariush-bahrami/gravity.optimizer>`__
|
|
218
|
-
|
|
219
|
-
| AdaSmooth | *An Adaptive Learning Rate Method based on Effective Ratio* |
|
|
220
|
-
|
|
221
|
-
| SRMM | *Stochastic regularized majorization-minimization with weakly convex and multi-convex surrogates* | `github <https://github.com/HanbaekLyu/SRMM>`__
|
|
222
|
-
|
|
223
|
-
| AvaGrad | *Domain-independent Dominance of Adaptive Methods* | `github <https://github.com/lolemacs/avagrad>`__
|
|
224
|
-
|
|
225
|
-
| PCGrad | *Gradient Surgery for Multi-Task Learning* | `github <https://github.com/tianheyu927/PCGrad>`__
|
|
226
|
-
|
|
227
|
-
| AMSGrad | *On the Convergence of Adam and Beyond* |
|
|
228
|
-
|
|
229
|
-
| Lookahead | *k steps forward, 1 step back* | `github <https://github.com/pytorch/examples/tree/main/imagenet>`__
|
|
230
|
-
|
|
231
|
-
| PNM | *Manipulating Stochastic Gradient Noise to Improve Generalization* | `github <https://github.com/zeke-xie/Positive-Negative-Momentum>`__
|
|
232
|
-
|
|
233
|
-
| GC | *Gradient Centralization* | `github <https://github.com/Yonghongwei/Gradient-Centralization>`__
|
|
234
|
-
|
|
235
|
-
| AGC | *Adaptive Gradient Clipping* | `github <https://github.com/deepmind/deepmind-research/tree/master/nfnets>`__
|
|
236
|
-
|
|
237
|
-
| Stable WD | *Understanding and Scheduling Weight Decay* | `github <https://github.com/zeke-xie/stable-weight-decay-regularization>`__
|
|
238
|
-
|
|
239
|
-
| Softplus T | *Calibrating the Adaptive Learning Rate to Improve Convergence of ADAM* |
|
|
240
|
-
|
|
241
|
-
| Un-tuned w/u | *On the adequacy of untuned warmup for adaptive optimization* |
|
|
242
|
-
|
|
243
|
-
| Norm Loss | *An efficient yet effective regularization method for deep neural networks* |
|
|
244
|
-
|
|
245
|
-
| AdaShift | *Decorrelation and Convergence of Adaptive Learning Rate Methods* | `github <https://github.com/MichaelKonobeev/adashift>`__
|
|
246
|
-
|
|
247
|
-
| AdaDelta | *An Adaptive Learning Rate Method* |
|
|
248
|
-
|
|
249
|
-
| Amos | *An Adam-style Optimizer with Adaptive Weight Decay towards Model-Oriented Scale* | `github <https://github.com/google-research/jestimator>`__
|
|
250
|
-
|
|
251
|
-
| SignSGD | *Compressed Optimisation for Non-Convex Problems* | `github <https://github.com/jxbz/signSGD>`__
|
|
252
|
-
|
|
253
|
-
| Sophia | *A Scalable Stochastic Second-order Optimizer for Language Model Pre-training* | `github <https://github.com/Liuhong99/Sophia>`__
|
|
254
|
-
|
|
255
|
-
| Prodigy | *An Expeditiously Adaptive Parameter-Free Learner* | `github <https://github.com/konstmish/prodigy>`__
|
|
256
|
-
|
|
257
|
-
| PAdam | *Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep Neural Networks* | `github <https://github.com/uclaml/Padam>`__
|
|
258
|
-
|
|
259
|
-
| LOMO | *Full Parameter Fine-tuning for Large Language Models with Limited Resources* | `github <https://github.com/OpenLMLab/LOMO>`__
|
|
260
|
-
|
|
136
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
137
|
+
| Optimizer | Description | Official Code | Paper | Citation |
|
|
138
|
+
+==============+===================================================================================================+==========================================================================================+===============================================================================================+======================================================================================================================+
|
|
139
|
+
| AdaBelief | *Adapting Step-sizes by the Belief in Observed Gradients* | `github <https://github.com/juntang-zhuang/Adabelief-Optimizer>`__ | `https://arxiv.org/abs/2010.07468 <https://arxiv.org/abs/2010.07468>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv201007468Z/exportcitation>`__ |
|
|
140
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
141
|
+
| AdaBound | *Adaptive Gradient Methods with Dynamic Bound of Learning Rate* | `github <https://github.com/Luolc/AdaBound/blob/master/adabound/adabound.py>`__ | `https://openreview.net/forum?id=Bkg3g2R9FX <https://openreview.net/forum?id=Bkg3g2R9FX>`__ | `cite <https://github.com/Luolc/AdaBound#citing>`__ |
|
|
142
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
143
|
+
| AdaHessian | *An Adaptive Second Order Optimizer for Machine Learning* | `github <https://github.com/amirgholami/adahessian>`__ | `https://arxiv.org/abs/2006.00719 <https://arxiv.org/abs/2006.00719>`__ | `cite <https://github.com/amirgholami/adahessian#citation>`__ |
|
|
144
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
145
|
+
| AdamD | *Improved bias-correction in Adam* | | `https://arxiv.org/abs/2110.10828 <https://arxiv.org/abs/2110.10828>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv211010828S/exportcitation>`__ |
|
|
146
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
147
|
+
| AdamP | *Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights* | `github <https://github.com/clovaai/AdamP>`__ | `https://arxiv.org/abs/2006.08217 <https://arxiv.org/abs/2006.08217>`__ | `cite <https://github.com/clovaai/AdamP#how-to-cite>`__ |
|
|
148
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
149
|
+
| diffGrad | *An Optimization Method for Convolutional Neural Networks* | `github <https://github.com/shivram1987/diffGrad>`__ | `https://arxiv.org/abs/1909.11015v3 <https://arxiv.org/abs/1909.11015v3>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190911015D/exportcitation>`__ |
|
|
150
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
151
|
+
| MADGRAD | *A Momentumized, Adaptive, Dual Averaged Gradient Method for Stochastic* | `github <https://github.com/facebookresearch/madgrad>`__ | `https://arxiv.org/abs/2101.11075 <https://arxiv.org/abs/2101.11075>`__ | `cite <https://github.com/facebookresearch/madgrad#tech-report>`__ |
|
|
152
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
153
|
+
| RAdam | *On the Variance of the Adaptive Learning Rate and Beyond* | `github <https://github.com/LiyuanLucasLiu/RAdam>`__ | `https://arxiv.org/abs/1908.03265 <https://arxiv.org/abs/1908.03265>`__ | `cite <https://github.com/LiyuanLucasLiu/RAdam#citation>`__ |
|
|
154
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
155
|
+
| Ranger | *a synergistic optimizer combining RAdam and LookAhead, and now GC in one optimizer* | `github <https://github.com/lessw2020/Ranger-Deep-Learning-Optimizer>`__ | `https://bit.ly/3zyspC3 <https://bit.ly/3zyspC3>`__ | `cite <https://github.com/lessw2020/Ranger-Deep-Learning-Optimizer#citing-this-work>`__ |
|
|
156
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
157
|
+
| Ranger21 | *a synergistic deep learning optimizer* | `github <https://github.com/lessw2020/Ranger21>`__ | `https://arxiv.org/abs/2106.13731 <https://arxiv.org/abs/2106.13731>`__ | `cite <https://github.com/lessw2020/Ranger21#referencing-this-work>`__ |
|
|
158
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
159
|
+
| Lamb | *Large Batch Optimization for Deep Learning* | `github <https://github.com/cybertronai/pytorch-lamb>`__ | `https://arxiv.org/abs/1904.00962 <https://arxiv.org/abs/1904.00962>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190400962Y/exportcitation>`__ |
|
|
160
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
161
|
+
| Shampoo | *Preconditioned Stochastic Tensor Optimization* | `github <https://github.com/moskomule/shampoo.pytorch>`__ | `https://arxiv.org/abs/1802.09568 <https://arxiv.org/abs/1802.09568>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180209568G/exportcitation>`__ |
|
|
162
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
163
|
+
| Nero | *Learning by Turning: Neural Architecture Aware Optimisation* | `github <https://github.com/jxbz/nero>`__ | `https://arxiv.org/abs/2102.07227 <https://arxiv.org/abs/2102.07227>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210207227L/exportcitation>`__ |
|
|
164
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
165
|
+
| Adan | *Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models* | `github <https://github.com/sail-sg/Adan>`__ | `https://arxiv.org/abs/2208.06677 <https://arxiv.org/abs/2208.06677>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv220806677X/exportcitation>`__ |
|
|
166
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
167
|
+
| Adai | *Disentangling the Effects of Adaptive Learning Rate and Momentum* | `github <https://github.com/zeke-xie/adaptive-inertia-adai>`__ | `https://arxiv.org/abs/2006.15815 <https://arxiv.org/abs/2006.15815>`__ | `cite <https://github.com/zeke-xie/adaptive-inertia-adai#citing>`__ |
|
|
168
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
169
|
+
| SAM | *Sharpness-Aware Minimization* | `github <https://github.com/davda54/sam>`__ | `https://arxiv.org/abs/2010.01412 <https://arxiv.org/abs/2010.01412>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv201001412F/exportcitation>`__ |
|
|
170
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
171
|
+
| ASAM | *Adaptive Sharpness-Aware Minimization* | `github <https://github.com/davda54/sam>`__ | `https://arxiv.org/abs/2102.11600 <https://arxiv.org/abs/2102.11600>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210211600K/exportcitation>`__ |
|
|
172
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
173
|
+
| GSAM | *Surrogate Gap Guided Sharpness-Aware Minimization* | `github <https://github.com/juntang-zhuang/GSAM>`__ | `https://openreview.net/pdf?id=edONMAnhLu- <https://openreview.net/pdf?id=edONMAnhLu->`__ | `cite <https://github.com/juntang-zhuang/GSAM#citation>`__ |
|
|
174
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
175
|
+
| D-Adaptation | *Learning-Rate-Free Learning by D-Adaptation* | `github <https://github.com/facebookresearch/dadaptation>`__ | `https://arxiv.org/abs/2301.07733 <https://arxiv.org/abs/2301.07733>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2023arXiv230107733D/exportcitation>`__ |
|
|
176
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
177
|
+
| AdaFactor | *Adaptive Learning Rates with Sublinear Memory Cost* | `github <https://github.com/DeadAt0m/adafactor-pytorch>`__ | `https://arxiv.org/abs/1804.04235 <https://arxiv.org/abs/1804.04235>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180404235S/exportcitation>`__ |
|
|
178
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
179
|
+
| Apollo | *An Adaptive Parameter-wise Diagonal Quasi-Newton Method for Nonconvex Stochastic Optimization* | `github <https://github.com/XuezheMax/apollo>`__ | `https://arxiv.org/abs/2009.13586 <https://arxiv.org/abs/2009.13586>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv200913586M/exportcitation>`__ |
|
|
180
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
181
|
+
| NovoGrad | *Stochastic Gradient Methods with Layer-wise Adaptive Moments for Training of Deep Networks* | `github <https://github.com/lonePatient/NovoGrad-pytorch>`__ | `https://arxiv.org/abs/1905.11286 <https://arxiv.org/abs/1905.11286>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190511286G/exportcitation>`__ |
|
|
182
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
183
|
+
| Lion | *Symbolic Discovery of Optimization Algorithms* | `github <https://github.com/google/automl/tree/master/lion>`__ | `https://arxiv.org/abs/2302.06675 <https://arxiv.org/abs/2302.06675>`__ | `cite <https://github.com/google/automl/tree/master/lion#citation>`__ |
|
|
184
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
185
|
+
| Ali-G | *Adaptive Learning Rates for Interpolation with Gradients* | `github <https://github.com/oval-group/ali-g>`__ | `https://arxiv.org/abs/1906.05661 <https://arxiv.org/abs/1906.05661>`__ | `cite <https://github.com/oval-group/ali-g#adaptive-learning-rates-for-interpolation-with-gradients>`__ |
|
|
186
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
187
|
+
| SM3 | *Memory-Efficient Adaptive Optimization* | `github <https://github.com/google-research/google-research/tree/master/sm3>`__ | `https://arxiv.org/abs/1901.11150 <https://arxiv.org/abs/1901.11150>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190111150A/exportcitation>`__ |
|
|
188
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
189
|
+
| AdaNorm | *Adaptive Gradient Norm Correction based Optimizer for CNNs* | `github <https://github.com/shivram1987/AdaNorm>`__ | `https://arxiv.org/abs/2210.06364 <https://arxiv.org/abs/2210.06364>`__ | `cite <https://github.com/shivram1987/AdaNorm/tree/main#citation>`__ |
|
|
190
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
191
|
+
| RotoGrad | *Gradient Homogenization in Multitask Learning* | `github <https://github.com/adrianjav/rotograd>`__ | `https://openreview.net/pdf?id=T8wHz4rnuGL <https://openreview.net/pdf?id=T8wHz4rnuGL>`__ | `cite <https://github.com/adrianjav/rotograd#citing>`__ |
|
|
192
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
193
|
+
| A2Grad | *Optimal Adaptive and Accelerated Stochastic Gradient Descent* | `github <https://github.com/severilov/A2Grad_optimizer>`__ | `https://arxiv.org/abs/1810.00553 <https://arxiv.org/abs/1810.00553>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv181000553D/exportcitation>`__ |
|
|
194
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
195
|
+
| AccSGD | *Accelerating Stochastic Gradient Descent For Least Squares Regression* | `github <https://github.com/rahulkidambi/AccSGD>`__ | `https://arxiv.org/abs/1704.08227 <https://arxiv.org/abs/1704.08227>`__ | `cite <https://github.com/rahulkidambi/AccSGD#citation>`__ |
|
|
196
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
197
|
+
| SGDW | *Decoupled Weight Decay Regularization* | `github <https://github.com/loshchil/AdamW-and-SGDW>`__ | `https://arxiv.org/abs/1711.05101 <https://arxiv.org/abs/1711.05101>`__ | `cite <https://github.com/loshchil/AdamW-and-SGDW#contact>`__ |
|
|
198
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
199
|
+
| ASGD | *Adaptive Gradient Descent without Descent* | `github <https://github.com/ymalitsky/adaptive_GD>`__ | `https://arxiv.org/abs/1910.09529 <https://arxiv.org/abs/1910.09529>`__ | `cite <https://github.com/ymalitsky/adaptive_GD#reference>`__ |
|
|
200
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
201
|
+
| Yogi | *Adaptive Methods for Nonconvex Optimization* | | `NIPS 2018 <https://papers.nips.cc/paper/8186-adaptive-methods-for-nonconvex-optimization>`__ | `cite <https://proceedings.neurips.cc/paper_files/paper/2018/hash/90365351ccc7437a1309dc64e4db32a3-Abstract.html>`__ |
|
|
202
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
203
|
+
| SWATS | *Improving Generalization Performance by Switching from Adam to SGD* | | `https://arxiv.org/abs/1712.07628 <https://arxiv.org/abs/1712.07628>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2017arXiv171207628S/exportcitation>`__ |
|
|
204
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
205
|
+
| Fromage | *On the distance between two neural networks and the stability of learning* | `github <https://github.com/jxbz/fromage>`__ | `https://arxiv.org/abs/2002.03432 <https://arxiv.org/abs/2002.03432>`__ | `cite <https://github.com/jxbz/fromage#citation>`__ |
|
|
206
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
207
|
+
| MSVAG | *Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradients* | `github <https://github.com/lballes/msvag>`__ | `https://arxiv.org/abs/1705.07774 <https://arxiv.org/abs/1705.07774>`__ | `cite <https://github.com/lballes/msvag#citation>`__ |
|
|
208
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
209
|
+
| AdaMod | *An Adaptive and Momental Bound Method for Stochastic Learning* | `github <https://github.com/lancopku/AdaMod>`__ | `https://arxiv.org/abs/1910.12249 <https://arxiv.org/abs/1910.12249>`__ | `cite <https://github.com/lancopku/AdaMod#citation>`__ |
|
|
210
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
211
|
+
| AggMo | *Aggregated Momentum: Stability Through Passive Damping* | `github <https://github.com/AtheMathmo/AggMo>`__ | `https://arxiv.org/abs/1804.00325 <https://arxiv.org/abs/1804.00325>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180400325L/exportcitation>`__ |
|
|
212
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
213
|
+
| QHAdam | *Quasi-hyperbolic momentum and Adam for deep learning* | `github <https://github.com/facebookresearch/qhoptim>`__ | `https://arxiv.org/abs/1810.06801 <https://arxiv.org/abs/1810.06801>`__ | `cite <https://github.com/facebookresearch/qhoptim#reference>`__ |
|
|
214
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
215
|
+
| PID | *A PID Controller Approach for Stochastic Optimization of Deep Networks* | `github <https://github.com/tensorboy/PIDOptimizer>`__ | `CVPR 18 <http://www4.comp.polyu.edu.hk/~cslzhang/paper/CVPR18_PID.pdf>`__ | `cite <https://github.com/tensorboy/PIDOptimizer#citation>`__ |
|
|
216
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
217
|
+
| Gravity | *a Kinematic Approach on Optimization in Deep Learning* | `github <https://github.com/dariush-bahrami/gravity.optimizer>`__ | `https://arxiv.org/abs/2101.09192 <https://arxiv.org/abs/2101.09192>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210109192B/exportcitation>`__ |
|
|
218
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
219
|
+
| AdaSmooth | *An Adaptive Learning Rate Method based on Effective Ratio* | | `https://arxiv.org/abs/2204.00825v1 <https://arxiv.org/abs/2204.00825v1>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv220400825L/exportcitation>`__ |
|
|
220
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
221
|
+
| SRMM | *Stochastic regularized majorization-minimization with weakly convex and multi-convex surrogates* | `github <https://github.com/HanbaekLyu/SRMM>`__ | `https://arxiv.org/abs/2201.01652 <https://arxiv.org/abs/2201.01652>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv220101652L/exportcitation>`__ |
|
|
222
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
223
|
+
| AvaGrad | *Domain-independent Dominance of Adaptive Methods* | `github <https://github.com/lolemacs/avagrad>`__ | `https://arxiv.org/abs/1912.01823 <https://arxiv.org/abs/1912.01823>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv191201823S/exportcitation>`__ |
|
|
224
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
225
|
+
| PCGrad | *Gradient Surgery for Multi-Task Learning* | `github <https://github.com/tianheyu927/PCGrad>`__ | `https://arxiv.org/abs/2001.06782 <https://arxiv.org/abs/2001.06782>`__ | `cite <https://github.com/tianheyu927/PCGrad#reference>`__ |
|
|
226
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
227
|
+
| AMSGrad | *On the Convergence of Adam and Beyond* | | `https://openreview.net/pdf?id=ryQu7f-RZ <https://openreview.net/pdf?id=ryQu7f-RZ>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190409237R/exportcitation>`__ |
|
|
228
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
229
|
+
| Lookahead | *k steps forward, 1 step back* | `github <https://github.com/pytorch/examples/tree/main/imagenet>`__ | `https://arxiv.org/abs/1907.08610 <https://arxiv.org/abs/1907.08610>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190708610Z/exportcitation>`__ |
|
|
230
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
231
|
+
| PNM | *Manipulating Stochastic Gradient Noise to Improve Generalization* | `github <https://github.com/zeke-xie/Positive-Negative-Momentum>`__ | `https://arxiv.org/abs/2103.17182 <https://arxiv.org/abs/2103.17182>`__ | `cite <https://github.com/zeke-xie/Positive-Negative-Momentum#citing>`__ |
|
|
232
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
233
|
+
| GC | *Gradient Centralization* | `github <https://github.com/Yonghongwei/Gradient-Centralization>`__ | `https://arxiv.org/abs/2004.01461 <https://arxiv.org/abs/2004.01461>`__ | `cite <https://github.com/Yonghongwei/Gradient-Centralization#citation>`__ |
|
|
234
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
235
|
+
| AGC | *Adaptive Gradient Clipping* | `github <https://github.com/deepmind/deepmind-research/tree/master/nfnets>`__ | `https://arxiv.org/abs/2102.06171 <https://arxiv.org/abs/2102.06171>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210206171B/exportcitation>`__ |
|
|
236
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
237
|
+
| Stable WD | *Understanding and Scheduling Weight Decay* | `github <https://github.com/zeke-xie/stable-weight-decay-regularization>`__ | `https://arxiv.org/abs/2011.11152 <https://arxiv.org/abs/2011.11152>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2020arXiv201111152X/exportcitation>`__ |
|
|
238
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
239
|
+
| Softplus T | *Calibrating the Adaptive Learning Rate to Improve Convergence of ADAM* | | `https://arxiv.org/abs/1908.00700 <https://arxiv.org/abs/1908.00700>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190800700T/exportcitation>`__ |
|
|
240
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
241
|
+
| Un-tuned w/u | *On the adequacy of untuned warmup for adaptive optimization* | | `https://arxiv.org/abs/1910.04209 <https://arxiv.org/abs/1910.04209>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv191004209M/exportcitation>`__ |
|
|
242
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
243
|
+
| Norm Loss | *An efficient yet effective regularization method for deep neural networks* | | `https://arxiv.org/abs/2103.06583 <https://arxiv.org/abs/2103.06583>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2021arXiv210306583G/exportcitation>`__ |
|
|
244
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
245
|
+
| AdaShift | *Decorrelation and Convergence of Adaptive Learning Rate Methods* | `github <https://github.com/MichaelKonobeev/adashift>`__ | `https://arxiv.org/abs/1810.00143v4 <https://arxiv.org/abs/1810.00143v4>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv181000143Z/exportcitation>`__ |
|
|
246
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
247
|
+
| AdaDelta | *An Adaptive Learning Rate Method* | | `https://arxiv.org/abs/1212.5701v1 <https://arxiv.org/abs/1212.5701v1>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2012arXiv1212.5701Z/exportcitation>`__ |
|
|
248
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
249
|
+
| Amos | *An Adam-style Optimizer with Adaptive Weight Decay towards Model-Oriented Scale* | `github <https://github.com/google-research/jestimator>`__ | `https://arxiv.org/abs/2210.11693 <https://arxiv.org/abs/2210.11693>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2022arXiv221011693T/exportcitation>`__ |
|
|
250
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
251
|
+
| SignSGD | *Compressed Optimisation for Non-Convex Problems* | `github <https://github.com/jxbz/signSGD>`__ | `https://arxiv.org/abs/1802.04434 <https://arxiv.org/abs/1802.04434>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2018arXiv180204434B/exportcitation>`__ |
|
|
252
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
253
|
+
| Sophia | *A Scalable Stochastic Second-order Optimizer for Language Model Pre-training* | `github <https://github.com/Liuhong99/Sophia>`__ | `https://arxiv.org/abs/2305.14342 <https://arxiv.org/abs/2305.14342>`__ | `cite <https://github.com/Liuhong99/Sophia>`__ |
|
|
254
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
255
|
+
| Prodigy | *An Expeditiously Adaptive Parameter-Free Learner* | `github <https://github.com/konstmish/prodigy>`__ | `https://arxiv.org/abs/2306.06101 <https://arxiv.org/abs/2306.06101>`__ | `cite <https://github.com/konstmish/prodigy#how-to-cite>`__ |
|
|
256
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
257
|
+
| PAdam | *Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep Neural Networks* | `github <https://github.com/uclaml/Padam>`__ | `https://arxiv.org/abs/1806.06763 <https://arxiv.org/abs/1806.06763>`__ | `cite <https://github.com/uclaml/Padam#citation>`__ |
|
|
258
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
259
|
+
| LOMO | *Full Parameter Fine-tuning for Large Language Models with Limited Resources* | `github <https://github.com/OpenLMLab/LOMO>`__ | `https://arxiv.org/abs/2306.09782 <https://arxiv.org/abs/2306.09782>`__ | `cite <https://github.com/OpenLMLab/LOMO#citation>`__ |
|
|
260
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
261
|
+
| Tiger | *A Tight-fisted Optimizer, an optimizer that is extremely budget-conscious* | `github <https://github.com/bojone/tiger>`__ | | `cite <https://github.com/bojone/tiger/blob/main/README_en.md#citation>`__ |
|
|
262
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
263
|
+
| CAME | *Confidence-guided Adaptive Memory Efficient Optimization* | `github <https://github.com/huawei-noah/Pretrained-Language-Model/tree/master/CAME>`__ | `https://aclanthology.org/2023.acl-long.243/ <https://aclanthology.org/2023.acl-long.243/>`__ | `cite <https://github.com/huawei-noah/Pretrained-Language-Model/tree/master/CAME#citation>`__ |
|
|
264
|
+
+--------------+---------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
261
265
|
|
|
262
266
|
Supported LR Scheduler
|
|
263
267
|
----------------------
|
|
264
268
|
|
|
265
269
|
You can check the supported learning rate schedulers with below code.
|
|
266
270
|
|
|
267
|
-
::
|
|
271
|
+
.. code-block:: python
|
|
268
272
|
|
|
269
273
|
from pytorch_optimizer import get_supported_lr_schedulers
|
|
270
274
|
|
|
@@ -283,7 +287,7 @@ Supported Loss Function
|
|
|
283
287
|
|
|
284
288
|
You can check the supported loss functions with below code.
|
|
285
289
|
|
|
286
|
-
::
|
|
290
|
+
.. code-block:: python
|
|
287
291
|
|
|
288
292
|
from pytorch_optimizer import get_supported_loss_functions
|
|
289
293
|
|
|
@@ -304,6 +308,10 @@ You can check the supported loss functions with below code.
|
|
|
304
308
|
+---------------------+-------------------------------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
305
309
|
| Bi-Tempered | *The Principle of Unchanged Optimality in Reinforcement Learning Generalization* | | `https://arxiv.org/abs/1906.03361 <https://arxiv.org/abs/1906.03361>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2019arXiv190600336I/exportcitation>`__ |
|
|
306
310
|
+---------------------+-------------------------------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
311
|
+
| Tversky | *Tversky loss function for image segmentation using 3D fully convolutional deep networks* | | `https://arxiv.org/abs/1706.05721 <https://arxiv.org/abs/1706.05721>`__ | `cite <https://ui.adsabs.harvard.edu/abs/2017arXiv170605721S/exportcitation>`__ |
|
|
312
|
+
+---------------------+-------------------------------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
313
|
+
| Lovasz Hinge | *A tractable surrogate for the optimization of the intersection-over-union measure in neural networks* | `github <https://github.com/bermanmaxim/LovaszSoftmax>`__ | `https://arxiv.org/abs/1705.08790 <https://arxiv.org/abs/1705.08790>`__ | `cite <https://github.com/bermanmaxim/LovaszSoftmax#citation>`__ |
|
|
314
|
+
+---------------------+-------------------------------------------------------------------------------------------------------------------------+-----------------------------------------------------------------------------------+-----------------------------------------------------------------------------------------------+----------------------------------------------------------------------------------------------------------------------+
|
|
307
315
|
|
|
308
316
|
Useful Resources
|
|
309
317
|
----------------
|
|
@@ -434,21 +442,22 @@ Improved bias-correction in Adam
|
|
|
434
442
|
Adaptive Gradient Norm Correction
|
|
435
443
|
---------------------------------
|
|
436
444
|
|
|
437
|
-
| Correcting the norm of gradient in each iteration based on the adaptive training history of gradient norm.
|
|
445
|
+
| Correcting the norm of a gradient in each iteration based on the adaptive training history of gradient norm.
|
|
438
446
|
|
|
439
447
|
Citation
|
|
440
448
|
--------
|
|
441
449
|
|
|
442
|
-
Please cite original authors of optimization algorithms. You can easily find it
|
|
443
|
-
If you use this software, please cite it
|
|
450
|
+
Please cite the original authors of optimization algorithms. You can easily find it in the above table!
|
|
451
|
+
If you use this software, please cite it below. Or you can get it from "cite this repository" button.
|
|
444
452
|
|
|
445
453
|
::
|
|
446
454
|
|
|
447
|
-
@software{
|
|
455
|
+
@software{Kim_pytorch_optimizer_optimizer_2022,
|
|
448
456
|
author = {Kim, Hyeongchan},
|
|
449
|
-
month =
|
|
450
|
-
title = {{pytorch_optimizer: optimizer
|
|
451
|
-
|
|
457
|
+
month = jan,
|
|
458
|
+
title = {{pytorch_optimizer: optimizer & lr scheduler & loss function collections in PyTorch}},
|
|
459
|
+
url = {https://github.com/kozistr/pytorch_optimizer},
|
|
460
|
+
version = {2.11.0},
|
|
452
461
|
year = {2022}
|
|
453
462
|
}
|
|
454
463
|
|
|
@@ -462,9 +471,9 @@ Hyeongchan Kim / `@kozistr <http://kozistr.tech/about>`__
|
|
|
462
471
|
:target: https://pytorch-optimizers.readthedocs.io/en/latest/?badge=latest
|
|
463
472
|
.. |PyPI version| image:: https://badge.fury.io/py/pytorch-optimizer.svg
|
|
464
473
|
:target: https://badge.fury.io/py/pytorch-optimizer
|
|
465
|
-
.. |PyPi download| image:: https://pepy.tech/badge/pytorch-optimizer
|
|
474
|
+
.. |PyPi download| image:: https://static.pepy.tech/badge/pytorch-optimizer
|
|
466
475
|
:target: https://pepy.tech/project/pytorch-optimizer
|
|
467
|
-
.. |PyPi month download| image:: https://pepy.tech/badge/pytorch-optimizer/month
|
|
476
|
+
.. |PyPi month download| image:: https://static.pepy.tech/badge/pytorch-optimizer/month
|
|
468
477
|
:target: https://pepy.tech/project/pytorch-optimizer
|
|
469
478
|
.. |PyPI pyversions| image:: https://img.shields.io/pypi/pyversions/pytorch-optimizer.svg
|
|
470
479
|
:target: https://pypi.python.org/pypi/pytorch-optimizer/
|