Fix Yogi import (#875)

tensorflow · Jan 14, 2020 · 0a23fd2 · 0a23fd2
1 parent cdf9de5
commit 0a23fd2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow_addons/optimizers/__init__.py b/tensorflow_addons/optimizers/__init__.py
@@ -39,3 +39,4 @@
 from tensorflow_addons.optimizers.weight_decay_optimizers import SGDW
 from tensorflow_addons.optimizers.weight_decay_optimizers import (
     extend_with_decoupled_weight_decay)
+from tensorflow_addons.optimizers.yogi import Yogi