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

perf: 修改默认训练的epoch数

This commit is contained in:
nl 2021-12-08 14:50:34 +08:00
parent dd78547199
commit f2727be430
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_speech_model.py
View File

@ -50,5 +50,5 @@ ms = ModelSpeech(sm251, feat, max_label_length=64)
#ms.load_model('save_models/' + sm251.get_model_name() + '.model.h5') #ms.load_model('save_models/' + sm251.get_model_name() + '.model.h5')
ms.train_model(optimizer=opt, data_loader=train_data, 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()) ms.save_model('save_models/' + sm251.get_model_name())