diff --git a/examples/hi_xiaowen/s0/conf/ds_tcn.yaml b/examples/hi_xiaowen/s0/conf/ds_tcn.yaml index e511502..6ef4407 100644 --- a/examples/hi_xiaowen/s0/conf/ds_tcn.yaml +++ b/examples/hi_xiaowen/s0/conf/ds_tcn.yaml @@ -37,6 +37,7 @@ model: optim: adam optim_conf: lr: 0.001 + weight_decay: 0.0001 training_config: grad_clip: 5 diff --git a/examples/hi_xiaowen/s0/conf/tcn.yaml b/examples/hi_xiaowen/s0/conf/tcn.yaml index 517b94e..35ff1c0 100644 --- a/examples/hi_xiaowen/s0/conf/tcn.yaml +++ b/examples/hi_xiaowen/s0/conf/tcn.yaml @@ -37,6 +37,7 @@ model: optim: adam optim_conf: lr: 0.001 + weight_decay: 0.0001 training_config: grad_clip: 5