From ecf117ca9596e570dfb5155abbd8966b1e82a486 Mon Sep 17 00:00:00 2001 From: cneud Date: Tue, 26 Apr 2022 11:50:20 +0200 Subject: [PATCH] adapt to tf1.compat session mode in tf2 --- qurator/eynollah/eynollah.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/qurator/eynollah/eynollah.py b/qurator/eynollah/eynollah.py index f8d7d09..784f07f 100644 --- a/qurator/eynollah/eynollah.py +++ b/qurator/eynollah/eynollah.py @@ -20,10 +20,10 @@ import numpy as np os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3" stderr = sys.stderr sys.stderr = open(os.devnull, "w") -from keras import backend as K -from keras.models import load_model -sys.stderr = stderr import tensorflow as tf +from tensorflow.python.keras import backend as K +from tensorflow.keras.models load_model +sys.stderr = stderr tf.get_logger().setLevel("ERROR") warnings.filterwarnings("ignore") from scipy.signal import find_peaks