From 8ff34be5e4a78a4470fad941c5cacbe156fc4ce3 Mon Sep 17 00:00:00 2001
From: iperov <lepersorium@gmail.com>
Date: Fri, 1 Jan 2021 17:37:12 +0400
Subject: [PATCH] leras.nn : support for tf ver 1

---
 core/leras/nn.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/core/leras/nn.py b/core/leras/nn.py
index c4420e1..76750fb 100644
--- a/core/leras/nn.py
+++ b/core/leras/nn.py
@@ -76,15 +76,19 @@ class nn():
             if first_run:
                 io.log_info("Caching GPU kernels...")
 
-            #import tensorflow as tf
-            import tensorflow.compat.v1 as tf
-            
+            import tensorflow
+            if tensorflow.VERSION[0] == '2':
+                tf = tensorflow.compat.v1
+            else:
+                tf = tensorflow
+
             import logging
             # Disable tensorflow warnings
             tf_logger = logging.getLogger('tensorflow')
             tf_logger.setLevel(logging.ERROR)
             
-            tf.disable_v2_behavior()
+            if tensorflow.VERSION[0] == '2':
+                tf.disable_v2_behavior()
             nn.tf = tf
 
             # Initialize framework