update

philipperemy · Jun 19, 2021 · eb8bc2a · eb8bc2a
1 parent afeefad
commit eb8bc2a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/keract/keract.py b/keract/keract.py
@@ -8,7 +8,9 @@
 from tensorflow.keras.models import Model
 import tensorflow as tf
 
-tf.compat.v1.experimental.output_all_intermediates(True)
+
+if tf.__version__ == '2.5.0':
+    tf.compat.v1.experimental.output_all_intermediates(True)
 
 
 def is_placeholder(n):