qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-06-09 20:29:55 +02:00
Code Issues Projects Releases Wiki Activity

adapt to tf1.compat session mode in tf2

Browse source
This commit is contained in:
cneud 2022-04-26 11:50:20 +02:00
parent 568391ec4a
commit ecf117ca95
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
qurator/eynollah/eynollah.py
View file

@ -20,10 +20,10 @@ import numpy as np
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3" os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
stderr = sys.stderr stderr = sys.stderr
sys.stderr = open(os.devnull, "w") 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 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") tf.get_logger().setLevel("ERROR")
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
from scipy.signal import find_peaks from scipy.signal import find_peaks