mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-11 19:12:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

url for pretrained models

Browse Source
This commit is contained in:
glynpu 2023-03-16 16:49:05 +08:00
parent d5471c5284
commit 20f36efedd
4 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/himia/wuw/ctc_tdnn/README.md Normal file
View File

@ -0,0 +1,5 @@
# Pretrained models and releated logs/results.
## ctc tdnn baseline
https://huggingface.co/GuoLiyong/himia_ctc_tdnn_baseline

2
egs/himia/wuw/ctc_tdnn/asr_datamodule.py
View File

@ -83,7 +83,7 @@ class HiMiaWuwDataModule:
group.add_argument( group.add_argument(
"--max-duration", "--max-duration",
type=int, type=int,
default=6000.0, default=200.0,
help="Maximum pooled recordings duration (seconds) in a " help="Maximum pooled recordings duration (seconds) in a "
"single batch. You can reduce it if it causes CUDA OOM.", "single batch. You can reduce it if it causes CUDA OOM.",
) )

6
egs/himia/wuw/ctc_tdnn/train.py
View File

@ -17,11 +17,11 @@
""" """
Usage: Usage:
export CUDA_VISIBLE_DEVICES="0,1,2,3" export CUDA_VISIBLE_DEVICES="0"
./ctc_tdnn/train.py \ ./ctc_tdnn/train.py \
--exp-dir ./ctc_tdnn/exp \ --exp-dir ./ctc_tdnn/exp \
--world-size 4 \ --world-size 1 \
--max-duration 200 \ --max-duration 100 \
--num-epochs 20 --num-epochs 20
""" """

6
egs/himia/wuw/run_ctc_tdnn.sh
View File

@ -8,7 +8,8 @@ stop_stage=2
epoch=10 epoch=10
avg=1 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} epoch_avg=epoch_${epoch}-avg_${avg}
post_dir=${exp_dir}/post/${epoch_avg} post_dir=${exp_dir}/post/${epoch_avg}
@ -24,7 +25,8 @@ log() {
if [ $stage -le 0 ] && [ $stop_stage -ge 0 ]; then if [ $stage -le 0 ] && [ $stop_stage -ge 0 ]; then
log "Stage 0: Model training" log "Stage 0: Model training"
python ./ctc_tdnn/train.py \ python ./ctc_tdnn/train.py \
--num-epochs $epoch --num-epochs $epoch \
--max-duration $max_duration
fi fi
if [ $stage -le 1 ] && [ $stop_stage -ge 1 ]; then if [ $stage -le 1 ] && [ $stop_stage -ge 1 ]; then