From f2727be430850145c5d21c68fb7687e2da9ec03f Mon Sep 17 00:00:00 2001 From: nl <3210346136@qq.com> Date: Wed, 8 Dec 2021 14:50:34 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=E4=BF=AE=E6=94=B9=E9=BB=98=E8=AE=A4?= =?UTF-8?q?=E8=AE=AD=E7=BB=83=E7=9A=84epoch=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- train_speech_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_speech_model.py b/train_speech_model.py index 625e433..d1596e7 100644 --- 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())