Updated Imports for TF-2.0

- Updated TF 2.0 Keras backend import - Depreciated Native Keras implementation [ci skip]
drahnreb · Nov 17, 2019 · f830c4b · f830c4b
1 parent 7269f88
commit f830c4b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Mish/Keras/mish.py b/Mish/Keras/mish.py
@@ -1,5 +1,7 @@
 # Keras Implementation of Mish Activation Function.
 
+"""Update: 11/17/2019- Keras implementation is now depreciated, please move to TFKeras directory for the Tensorflow 2.0 Keras based implementation"""
+
 # Import Necessary Modules.
 from __future__ import absolute_import
 from __future__ import division

diff --git a/Mish/TFKeras/mish.py b/Mish/TFKeras/mish.py
@@ -3,7 +3,7 @@
 ## Import Necessary Modules
 import tensorflow as tf
 from tensorflow.keras.layers import Activation
-from tensorflow.keras.utils.generic_utils import get_custom_objects
+from tensorflow.keras.utils import get_custom_objects
 
 class Mish(Activation):
     '''