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

[kws] fix weight_decay key error

Browse Source
This commit is contained in:
Binbin Zhang 2021-12-04 15:15:01 +08:00
parent dfe8b2536b
commit 5bd9f00977
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',