feat: 增加save_step的数，避免大量模型文件产生

2021-08-10 17:14:34 +08:00 · 2021-08-10 17:14:34 +08:00 · 4ea884bfa7
parent ce9b1c1fa3
commit 4ea884bfa7
1 changed files with 1 additions and 1 deletions
--- a/train_mspeech.py
+++ b/train_mspeech.py
@ -64,6 +64,6 @@ else:
 ms = ModelSpeech(datapath)

 #ms.LoadModel(modelpath + 'speech_model251_e_0_step_327500.h5')
-ms.TrainModel(datapath, epoch = 50, batch_size = 16, save_step = 500)
+ms.TrainModel(datapath, epoch = 50, batch_size = 16, save_step = 5000)