[kws] use **kvargs for optim to reduce code (#25)

2021-12-06 11:23:44 +08:00 · 2021-12-06 11:23:44 +08:00 · a5a54782cc
commit a5a54782cc
parent 88444ab177
1 changed files with 1 additions and 3 deletions
--- a/kws/bin/train.py
+++ b/kws/bin/train.py
@ -195,9 +195,7 @@ def main():
        device = torch.device('cuda' if use_cuda else 'cpu')
        model = model.to(device)

-    optimizer = optim.Adam(model.parameters(),
-                           lr=configs['optim_conf']['lr'],
-                           weight_decay=configs['optim_conf'].get('weight_decay', 0))
+    optimizer = optim.Adam(model.parameters(), **configs['optim_conf'])
    scheduler = optim.lr_scheduler.ReduceLROnPlateau(
        optimizer,
        mode='min',