Skip to content

Commit 830f7e1

Browse files
committed
update: Ranger25 optimizer
1 parent 74dd8bc commit 830f7e1

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

pytorch_optimizer/optimizer/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
4040
from pytorch_optimizer.optimizer.dadapt import DAdaptAdaGrad, DAdaptAdam, DAdaptAdan, DAdaptLion, DAdaptSGD
4141
from pytorch_optimizer.optimizer.demo import DeMo
4242
from pytorch_optimizer.optimizer.diffgrad import DiffGrad
43+
from pytorch_optimizer.optimizer.experimental.ranger25 import Ranger25
4344
from pytorch_optimizer.optimizer.fadam import FAdam
4445
from pytorch_optimizer.optimizer.fp16 import DynamicLossScaler, SafeFP16Optimizer
4546
from pytorch_optimizer.optimizer.fromage import Fromage
@@ -284,6 +285,7 @@ def load_optimizer(optimizer: str) -> OPTIMIZER:
284285
MARS,
285286
SGDSaI,
286287
Grams,
288+
Ranger25,
287289
]
288290
OPTIMIZERS: Dict[str, OPTIMIZER] = {str(optimizer.__name__).lower(): optimizer for optimizer in OPTIMIZER_LIST}
289291

pytorch_optimizer/optimizer/experimental/__init__.py

Whitespace-only changes.

0 commit comments

Comments
 (0)