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

change wer-summary-{test_set_name} to wer-summary-{test_set_name}-{params.method}

Browse Source
This commit is contained in:
sathvik udupa 2023-05-03 11:38:32 +05:30
parent e234e4b9b0
commit 66fb375dbb
2 changed files with 17 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/mucs/ASR/conformer_ctc/decode.py
View File

@ -601,7 +601,7 @@ def save_results(
logging.info("Wrote detailed error stats to {}".format(errs_filename)) logging.info("Wrote detailed error stats to {}".format(errs_filename))
test_set_wers = sorted(test_set_wers.items(), key=lambda x: x[1]) test_set_wers = sorted(test_set_wers.items(), key=lambda x: x[1])
errs_info = params.exp_dir / f"wer-summary-{test_set_name}.txt" errs_info = params.exp_dir / f"wer-summary-{test_set_name}-{params.method}.txt"
with open(errs_info, "w") as f: with open(errs_info, "w") as f:
print("settings\tWER", file=f) print("settings\tWER", file=f)
for key, val in test_set_wers: for key, val in test_set_wers:

26
egs/mucs/ASR/run.sh
View File

@ -5,20 +5,26 @@ set -e
dataset='bn-en' dataset='bn-en'
datadir=data_"$dataset" datadir=data_"$dataset"
bpe=400 bpe=400
decode_method="nbest"
num_paths=20
./conformer_ctc/train.py \ # ./conformer_ctc/train.py \
--num-epochs 60 \ # --num-epochs 60 \
--max-duration 300 \ # --max-duration 300 \
--exp-dir ./conformer_ctc/exp_"$dataset"_bpe"$bpe" \ # --exp-dir ./conformer_ctc/exp_"$dataset"_bpe"$bpe" \
--manifest-dir $datadir/fbank \ # --manifest-dir $datadir/fbank \
--lang-dir $datadir/lang_bpe_"$bpe" \ # --lang-dir $datadir/lang_bpe_"$bpe" \
--enable-musan False \ # --enable-musan False \
./conformer_ctc/decode.py \ ./conformer_ctc/decode.py \
--epoch 60 \ --epoch 59 \
--avg 10 \ --avg 10 \
--manifest-dir $datadir/fbank \ --manifest-dir $datadir/fbank \
--exp-dir ./conformer_ctc/exp_"$dataset"_bpe"$bpe" \ --exp-dir ./conformer_ctc/exp_"$dataset"_bpe"$bpe" \
--max-duration 100 \ --max-duration 5 \
--lang-dir $datadir/lang_bpe_"$bpe" --lang-dir $datadir/lang_bpe_"$bpe" \
--lm-dir $datadir/"lm" \
--method $decode_method \
--num-paths $num_paths \