Skip to content

Commit 9b3c3d2

Browse files
committed
feature: DiffRGrad optimizer
1 parent fd67b50 commit 9b3c3d2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pytorch_optimizer/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from pytorch_optimizer.agc import agc
77
from pytorch_optimizer.chebyshev_schedule import get_chebyshev_schedule
88
from pytorch_optimizer.diffgrad import DiffGrad
9+
from pytorch_optimizer.diffrgrad import DiffRGrad
910
from pytorch_optimizer.gc import centralize_gradient
1011
from pytorch_optimizer.lookahead import Lookahead
1112
from pytorch_optimizer.madgrad import MADGRAD
@@ -15,4 +16,4 @@
1516
from pytorch_optimizer.sam import SAM
1617
from pytorch_optimizer.sgdp import SGDP
1718

18-
__VERSION__ = '0.0.8'
19+
__VERSION__ = '0.0.9'

0 commit comments

Comments
 (0)