update

2021-12-06 12:09:43 +08:00 · 2021-12-06 12:09:43 +08:00 · 0140e15ad8
commit 0140e15ad8
parent dc596b41a1
3 changed files with 4 additions and 3 deletions
--- a/examples/speechcommand_v1/s0/conf/mdtc.yaml
+++ b/examples/speechcommand_v1/s0/conf/mdtc.yaml
@ -42,7 +42,7 @@ model:

 optim: adam
 optim_conf:
-    lr: 0.0002
+    lr: 0.001
    weight_decay: 0.00005

 training_config:
--- a/examples/speechcommand_v1/s0/run.sh
+++ b/examples/speechcommand_v1/s0/run.sh
@ -6,7 +6,7 @@

 export CUDA_VISIBLE_DEVICES="0"

-stage=1
+stage=2
 stop_stage=4
 num_keywords=11

--- a/kws/bin/train.py
+++ b/kws/bin/train.py
@ -196,7 +196,8 @@ def main():
        model = model.to(device)

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