From 20f36efedd97d2b3fc0752da367c7eae59cdbec5 Mon Sep 17 00:00:00 2001
From: glynpu <guonpu@qq.com>
Date: Thu, 16 Mar 2023 16:49:05 +0800
Subject: [PATCH] url for pretrained models

---
 egs/himia/wuw/ctc_tdnn/README.md         | 5 +++++
 egs/himia/wuw/ctc_tdnn/asr_datamodule.py | 2 +-
 egs/himia/wuw/ctc_tdnn/train.py          | 6 +++---
 egs/himia/wuw/run_ctc_tdnn.sh            | 6 ++++--
 4 files changed, 13 insertions(+), 6 deletions(-)
 create mode 100644 egs/himia/wuw/ctc_tdnn/README.md

diff --git a/egs/himia/wuw/ctc_tdnn/README.md b/egs/himia/wuw/ctc_tdnn/README.md
new file mode 100644
index 000000000..6eeb9161f
--- /dev/null
+++ b/egs/himia/wuw/ctc_tdnn/README.md
@@ -0,0 +1,5 @@
+# Pretrained models and releated logs/results.
+
+## ctc tdnn baseline
+
+https://huggingface.co/GuoLiyong/himia_ctc_tdnn_baseline
diff --git a/egs/himia/wuw/ctc_tdnn/asr_datamodule.py b/egs/himia/wuw/ctc_tdnn/asr_datamodule.py
index c02a8e634..785e06a7c 100644
--- a/egs/himia/wuw/ctc_tdnn/asr_datamodule.py
+++ b/egs/himia/wuw/ctc_tdnn/asr_datamodule.py
@@ -83,7 +83,7 @@ class HiMiaWuwDataModule:
         group.add_argument(
             "--max-duration",
             type=int,
-            default=6000.0,
+            default=200.0,
             help="Maximum pooled recordings duration (seconds) in a "
             "single batch. You can reduce it if it causes CUDA OOM.",
         )
diff --git a/egs/himia/wuw/ctc_tdnn/train.py b/egs/himia/wuw/ctc_tdnn/train.py
index fd9d42cad..61842de79 100755
--- a/egs/himia/wuw/ctc_tdnn/train.py
+++ b/egs/himia/wuw/ctc_tdnn/train.py
@@ -17,11 +17,11 @@
 
 """
 Usage:
-  export CUDA_VISIBLE_DEVICES="0,1,2,3"
+  export CUDA_VISIBLE_DEVICES="0"
   ./ctc_tdnn/train.py \
      --exp-dir ./ctc_tdnn/exp \
-     --world-size 4 \
-     --max-duration 200 \
+     --world-size 1 \
+     --max-duration 100 \
      --num-epochs 20
 """
 
diff --git a/egs/himia/wuw/run_ctc_tdnn.sh b/egs/himia/wuw/run_ctc_tdnn.sh
index fdf5ec1db..8a65d9c54 100644
--- a/egs/himia/wuw/run_ctc_tdnn.sh
+++ b/egs/himia/wuw/run_ctc_tdnn.sh
@@ -8,7 +8,8 @@ stop_stage=2
 
 epoch=10
 avg=1
-exp_dir=./ctc_tdnn/exp/
+max_duration=150
+exp_dir=./ctc_tdnn/exp_max_duration_${max_duration}/
 epoch_avg=epoch_${epoch}-avg_${avg}
 post_dir=${exp_dir}/post/${epoch_avg}
 
@@ -24,7 +25,8 @@ log() {
 if [ $stage -le 0 ] && [ $stop_stage -ge 0 ]; then
   log "Stage 0: Model training"
   python ./ctc_tdnn/train.py \
-    --num-epochs $epoch
+    --num-epochs $epoch \
+    --max-duration $max_duration
 fi
 
 if [ $stage -le 1 ] && [ $stop_stage -ge 1 ]; then