OrgLion-ML
/
ASRT_SpeechRecognition
mirror of https://github.com/nl8590687/ASRT_SpeechRecognition.git
1
0
Fork 0
Code Issues Releases Wiki Activity

自动创建保存模型的子目录

This commit is contained in:
nl 2020-08-10 16:35:40 +08:00
parent c1eb8862fa
commit 87a0ca090e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train_mspeech.py
View File

@ -12,7 +12,7 @@ import tensorflow as tf
from keras.backend.tensorflow_backend import set_session from keras.backend.tensorflow_backend import set_session
from SpeechModel251 import ModelSpeech from SpeechModel251 import ModelSpeech, ModelName
os.environ["CUDA_VISIBLE_DEVICES"] = "0" os.environ["CUDA_VISIBLE_DEVICES"] = "0"
#进行配置，使用95%的GPU #进行配置，使用95%的GPU
@ -29,6 +29,7 @@ modelpath = 'model_speech'
if(not os.path.exists(modelpath)): # 判断保存模型的目录是否存在 if(not os.path.exists(modelpath)): # 判断保存模型的目录是否存在
os.makedirs(modelpath) # 如果不存在，就新建一个，避免之后保存模型的时候炸掉 os.makedirs(modelpath) # 如果不存在，就新建一个，避免之后保存模型的时候炸掉
os.makedirs(modelpath + '/m' + ModelName)
system_type = plat.system() # 由于不同的系统的文件路径表示不一样，需要进行判断 system_type = plat.system() # 由于不同的系统的文件路径表示不一样，需要进行判断
if(system_type == 'Windows'): if(system_type == 'Windows'):