修复部分代码对tf2.0的兼容问题

2020-05-30 12:52:53 +08:00 · 2020-05-30 12:52:53 +08:00 · 74d5e40dea
parent 4dad3e73e1
commit 74d5e40dea
1 changed files with 2 additions and 2 deletions
--- a/SpeechModel251.py
+++ b/SpeechModel251.py
@ -323,8 +323,8 @@ class ModelSpeech(): # 语音模型类
 		r = K.ctc_decode(base_pred, in_len, greedy = True, beam_width=100, top_paths=1)
 		
 		#print('r', r)
-		r1 = r[0][0].eval(session=tf.Session())
-		tf.reset_default_graph()
+		r1 = r[0][0].eval(session=tf.compat.v1.Session())
+		tf.compat.v1.reset_default_graph()
 		return r1[0]
 	
 	def RecognizeSpeech(self, wavsignal, fs):