switch model to 251

nl8590687 · Jul 27, 2018 · 201c018 · 201c018
1 parent c09853f
commit 201c018
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/asrserver.py b/asrserver.py
@@ -8,13 +8,13 @@
 import http.server
 import urllib
 import keras
-from SpeechModel25 import ModelSpeech
+from SpeechModel251 import ModelSpeech
 from LanguageModel import ModelLanguage
 
 datapath = 'data/'
 modelpath = 'model_speech/'
 ms = ModelSpeech(datapath)
-ms.LoadModel(modelpath + 'm25/speech_model25_e_0_step_545500.model')
+ms.LoadModel(modelpath + 'm251/speech_model251_e_0_step_12000.model')
 
 ml = ModelLanguage('model_language')
 ml.LoadModel()

diff --git a/test_mspeech.py b/test_mspeech.py
@@ -12,7 +12,7 @@
 from keras.backend.tensorflow_backend import set_session
 
 
-from SpeechModel25 import ModelSpeech
+from SpeechModel251 import ModelSpeech
 
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
@@ -44,7 +44,7 @@
 
 ms = ModelSpeech(datapath)
 
-ms.LoadModel(modelpath + 'm25/speech_model25_e_0_step_42500.model')
+ms.LoadModel(modelpath + 'm251/speech_model251_e_0_step_42500.model')
 
 ms.TestModel(datapath, str_dataset='test', data_count = 128, out_report = True)
 

diff --git a/train_mspeech.py b/train_mspeech.py
@@ -12,7 +12,7 @@
 from keras.backend.tensorflow_backend import set_session
 
 
-from SpeechModel25 import ModelSpeech
+from SpeechModel251 import ModelSpeech
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 #进行配置，使用95%的GPU
@@ -43,7 +43,7 @@
 
 ms = ModelSpeech(datapath)
 
-#ms.LoadModel(modelpath + 'speech_model24_e_0_step_327500.model')
+#ms.LoadModel(modelpath + 'speech_model251_e_0_step_327500.model')
 ms.TrainModel(datapath, epoch = 50, batch_size = 16, save_step = 500)