From 8cfd4ed4f2c7ed9bf4c161bc931b604185768070 Mon Sep 17 00:00:00 2001
From: Binbin Zhang <binbzha@qq.com>
Date: Sat, 4 Dec 2021 17:19:46 +0800
Subject: [PATCH] [kws] fix weight_decay key error (#23)

---
 kws/bin/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kws/bin/train.py b/kws/bin/train.py
index 3497322..a6c9472 100644
--- 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',