AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix intent bug with initial learning rate != 0.001

Browse Source
This commit is contained in:
blessyyyu 2022-04-14 15:49:41 +08:00
parent 98768c7976
commit 2be4405b38
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
kws/bin/train.py
View File

@ -158,7 +158,7 @@ def main():
start_epoch = infos.get('epoch', -1) + 1 start_epoch = infos.get('epoch', -1) + 1
cv_loss = infos.get('cv_loss', 0.0) cv_loss = infos.get('cv_loss', 0.0)
# get the last epoch lr # get the last epoch lr
lr_last_epoch = infos.get('lr', 0.001) lr_last_epoch = infos.get('lr', configs['optim_conf']['lr'])
configs['optim_conf']['lr'] = lr_last_epoch configs['optim_conf']['lr'] = lr_last_epoch
model_dir = args.model_dir model_dir = args.model_dir
writer = None writer = None