AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
jingyong hou 2021-12-06 12:09:43 +08:00
parent dc596b41a1
commit 0140e15ad8
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/speechcommand_v1/s0/conf/mdtc.yaml
View File

@ -42,7 +42,7 @@ model:
optim: adam optim: adam
optim_conf: optim_conf:
lr: 0.0002 lr: 0.001
weight_decay: 0.00005 weight_decay: 0.00005
training_config: training_config:

2
examples/speechcommand_v1/s0/run.sh
View File

@ -6,7 +6,7 @@
export CUDA_VISIBLE_DEVICES="0" export CUDA_VISIBLE_DEVICES="0"
stage=1 stage=2
stop_stage=4 stop_stage=4
num_keywords=11 num_keywords=11

3
kws/bin/train.py
View File

@ -196,7 +196,8 @@ def main():
model = model.to(device) model = model.to(device)
optimizer = optim.Adam(model.parameters(), 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( scheduler = optim.lr_scheduler.ReduceLROnPlateau(
optimizer, optimizer,
mode='min', mode='min',