From f50f7bc81abe9305d18faa2f291deadf4f7a12c0 Mon Sep 17 00:00:00 2001 From: Yizheng Huang Date: Mon, 21 Jan 2019 20:28:43 +0800 Subject: [PATCH] Update SpeechModel251.py --- SpeechModel251.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/SpeechModel251.py b/SpeechModel251.py index bf77942..9bf2b03 100644 --- a/SpeechModel251.py +++ b/SpeechModel251.py @@ -197,8 +197,11 @@ class ModelSpeech(): # 语音模型类 ''' 保存模型参数 ''' - self._model.save_weights(filename+comment+'.model') + self._model.save_weights(filename + comment + '.model') self.base_model.save_weights(filename + comment + '.model.base') + # 需要安装 hdf5 模块 + self._model.save(filename + comment + '.h5') + self.base_model.save(filename + comment + '.base.h5') f = open('step'+ModelName+'.txt','w') f.write(filename+comment) f.close()