Skip to content

Commit aa3d7b0

Browse files
committed
Add imports to __init__.py
1 parent ec55b29 commit aa3d7b0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tensorflow_addons/optimizers/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,3 +19,7 @@
1919
from __future__ import print_function
2020

2121
from tensorflow_addons.optimizers.lazy_adam import LazyAdam
22+
from tensorflow_addons.optimizers.weight_decay_optimizers import AdamWOptimizer
23+
from tensorflow_addons.optimizers.weight_decay_optimizers import SGDWOptimizer
24+
from tensorflow_addons.optimizers.weight_decay_optimizers import (
25+
extend_with_decoupled_weight_decay)

0 commit comments

Comments
 (0)