Skip to content
This repository was archived by the owner on Apr 19, 2023. It is now read-only.

Commit 4ebc5e8

Browse files
Add support for ranger optimizers
1 parent 45df620 commit 4ebc5e8

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed
+2-1
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
from .adam import Adam
2-
from .annealed_adam import AnnealedAdam
2+
from .annealed_adam import AnnealedAdam
3+
from .ranger import Ranger, RangerQH, RangerVA
+8
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
# easy support for additional ranger optimizers from
2+
# https://github.com/lessw2020/Ranger-Deep-Learning-Optimizer
3+
try:
4+
from ranger import Ranger, RangerVA, RangerQH
5+
except ImportError:
6+
Ranger = None
7+
RangerVA = None
8+
RangerQH = None

0 commit comments

Comments
 (0)