diff --git a/train_mspeech.py b/train_mspeech.py index c4f1807..df31180 100644 --- 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)