url for pretrained models

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

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.",
         )

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
 """
 

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