diff --git a/asrserver.py b/asrserver.py index d97251e..7931064 100644 --- 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 index b7c918f..563741b 100644 --- a/test_mspeech.py +++ b/test_mspeech.py @@ -12,7 +12,7 @@ import tensorflow as tf 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 @@ else: 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 index cdb4874..4748f2e 100644 --- a/train_mspeech.py +++ b/train_mspeech.py @@ -12,7 +12,7 @@ import tensorflow as tf 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 @@ else: 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)