perf: 修改默认训练的epoch数

2021-12-08 14:50:34 +08:00 · 2021-12-08 14:50:34 +08:00 · f2727be430
parent dd78547199
commit f2727be430
1 changed files with 1 additions and 1 deletions
--- a/train_speech_model.py
+++ b/train_speech_model.py
@ -50,5 +50,5 @@ ms = ModelSpeech(sm251, feat, max_label_length=64)

 #ms.load_model('save_models/' + sm251.get_model_name() + '.model.h5')
 ms.train_model(optimizer=opt, data_loader=train_data,
-    epochs=1, save_step=1, batch_size=16, last_epoch=0)
+    epochs=50, save_step=1, batch_size=16, last_epoch=0)
 ms.save_model('save_models/' + sm251.get_model_name())