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

[kws] fix weight_decay key error (#23)

Browse Source
This commit is contained in:
Binbin Zhang 2021-12-04 17:19:46 +08:00 committed by GitHub
parent dfe8b2536b
commit 8cfd4ed4f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

@ -197,7 +197,7 @@ def main():
optimizer = optim.Adam(model.parameters(), optimizer = optim.Adam(model.parameters(),
lr=configs['optim_conf']['lr'], lr=configs['optim_conf']['lr'],
weight_decay=configs['optim_conf']['weight_decay']) weight_decay=configs['optim_conf'].get('weight_decay', 0))
scheduler = optim.lr_scheduler.ReduceLROnPlateau( scheduler = optim.lr_scheduler.ReduceLROnPlateau(
optimizer, optimizer,
mode='min', mode='min',