From 0a23fd2dc41be4725da27e8f68ddfb71cef72bc1 Mon Sep 17 00:00:00 2001 From: Sean Morgan Date: Tue, 14 Jan 2020 11:15:04 -0500 Subject: [PATCH] Fix Yogi import (#875) --- tensorflow_addons/optimizers/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow_addons/optimizers/__init__.py b/tensorflow_addons/optimizers/__init__.py index 060d0f4a7e..17c08d742f 100644 --- 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