From 87a0ca090e0670891f719c0754a59d9f8ab1f8e6 Mon Sep 17 00:00:00 2001
From: nl <3210346136@qq.com>
Date: Mon, 10 Aug 2020 16:35:40 +0800
Subject: [PATCH] =?UTF-8?q?=E8=87=AA=E5=8A=A8=E5=88=9B=E5=BB=BA=E4=BF=9D?=
 =?UTF-8?q?=E5=AD=98=E6=A8=A1=E5=9E=8B=E7=9A=84=E5=AD=90=E7=9B=AE=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 train_mspeech.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/train_mspeech.py b/train_mspeech.py
index 2d99c6d..44bf0dc 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 SpeechModel251 import ModelSpeech
+from SpeechModel251 import ModelSpeech, ModelName
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 #进行配置，使用95%的GPU
@@ -29,6 +29,7 @@ modelpath = 'model_speech'
 
 if(not os.path.exists(modelpath)): # 判断保存模型的目录是否存在
 	os.makedirs(modelpath) # 如果不存在，就新建一个，避免之后保存模型的时候炸掉
+	os.makedirs(modelpath + '/m' + ModelName)
 
 system_type = plat.system() # 由于不同的系统的文件路径表示不一样，需要进行判断
 if(system_type == 'Windows'):