[kws] fix weight_decay key error (#23)

2021-12-04 17:19:46 +08:00 · 2021-12-04 17:19:46 +08:00 · 8cfd4ed4f2
commit 8cfd4ed4f2
parent dfe8b2536b
1 changed files with 1 additions and 1 deletions
--- a/kws/bin/train.py
+++ b/kws/bin/train.py
@ -197,7 +197,7 @@ def main():

    optimizer = optim.Adam(model.parameters(),
                           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(
        optimizer,
        mode='min',