Skip to content

Commit b8e71b1

Browse files
committed
update: optimizer utils
1 parent db70700 commit b8e71b1

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
@@ -16,5 +16,6 @@
1616
from pytorch_optimizer.ranger21 import Ranger21
1717
from pytorch_optimizer.sam import SAM
1818
from pytorch_optimizer.sgdp import SGDP
19+
from pytorch_optimizer.utils import get_optimizer_parameters, normalize_gradient, unit_norm
1920

20-
__VERSION__ = '0.1.0'
21+
__VERSION__ = '0.1.1'

0 commit comments

Comments
 (0)