From 92a4c19ffe9d3ec3af75dadbf051b18b86a8daec Mon Sep 17 00:00:00 2001
From: Binbin Zhang <binbzha@qq.com>
Date: Tue, 7 Dec 2021 10:36:38 +0800
Subject: [PATCH] [examples] use ds_tcn as default model (#34)

* [examples] use ds_tcn as default model

* fix scoring gpu id
---
 examples/hi_xiaowen/s0/README.md |  6 ++----
 examples/hi_xiaowen/s0/run.sh    | 13 +++++++------
 2 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/examples/hi_xiaowen/s0/README.md b/examples/hi_xiaowen/s0/README.md
index b88ad49..87ff5c3 100644
--- a/examples/hi_xiaowen/s0/README.md
+++ b/examples/hi_xiaowen/s0/README.md
@@ -4,9 +4,7 @@ FRRs with FAR fixed at once per hour:
 |------------------|-----------|-----------|------------|--------------|
 | GRU              | 203       | 80(avg30) | 0.088901   | 0.083827     |
 | TCN              | 134       | 80(avg30) | 0.023494   | 0.029884     |
-| DS_TCN           | 21        | 60        | 0.011559   | 0.014190     |
-| DS_TCN           | 21        | 80        | 0.010807   | 0.014754     |
-| DS_TCN           | 21        | 80(avg30) | 0.009867   | 0.014472     |
-| DS_TCN(spec_aug) | 21        | 80(avg30) | 0.029039   | 0.022648     |
+| DS_TCN           | 21        | 80(avg30) | 0.019641   | 0.018325     |
+| DS_TCN(spec_aug) | 21        | 80(avg30) | 0.029509   | 0.008928     |
 | MDTC             | 156       | 80(avg10) | 0.007142   | 0.005920     |
 | MDTC_Small       | 31        | 80(avg10) | 0.005357   | 0.005920     |
diff --git a/examples/hi_xiaowen/s0/run.sh b/examples/hi_xiaowen/s0/run.sh
index 7c97999..3f7967b 100755
--- a/examples/hi_xiaowen/s0/run.sh
+++ b/examples/hi_xiaowen/s0/run.sh
@@ -9,15 +9,15 @@ stage=0
 stop_stage=4
 num_keywords=2
 
-config=conf/mdtc_small.yaml
-norm_mean=false
-norm_var=false
+config=conf/ds_tcn.yaml
+norm_mean=true
+norm_var=true
 gpu_id=0
 
 checkpoint=
-dir=exp/mdtc_small
+dir=exp/ds_tcn
 
-num_average=10
+num_average=30
 score_checkpoint=$dir/avg_${num_average}.pt
 
 download_dir=./data/local # your data dir
@@ -82,6 +82,7 @@ if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
     --num_workers 8 \
     --num_keywords $num_keywords \
     --min_duration 50 \
+    --seed 666 \
     $cmvn_opts \
     ${checkpoint:+--checkpoint $checkpoint}
 fi
@@ -97,7 +98,7 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
   # Compute posterior score
   result_dir=$dir/test_$(basename $score_checkpoint)
   mkdir -p $result_dir
-  python kws/bin/score.py --gpu 1 \
+  python kws/bin/score.py --gpu $gpu_id \
     --config $dir/config.yaml \
     --test_data data/test/data.list \
     --batch_size 256 \