diff --git a/egs/mucs/ASR/conformer_ctc/decode.py b/egs/mucs/ASR/conformer_ctc/decode.py index 907fc6c8f..812e5db34 100755 --- a/egs/mucs/ASR/conformer_ctc/decode.py +++ b/egs/mucs/ASR/conformer_ctc/decode.py @@ -601,7 +601,7 @@ def save_results( logging.info("Wrote detailed error stats to {}".format(errs_filename)) 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: print("settings\tWER", file=f) for key, val in test_set_wers: diff --git a/egs/mucs/ASR/run.sh b/egs/mucs/ASR/run.sh index 830c30e08..65dcea1f3 100755 --- a/egs/mucs/ASR/run.sh +++ b/egs/mucs/ASR/run.sh @@ -5,20 +5,26 @@ set -e dataset='bn-en' datadir=data_"$dataset" bpe=400 +decode_method="nbest" +num_paths=20 -./conformer_ctc/train.py \ - --num-epochs 60 \ - --max-duration 300 \ - --exp-dir ./conformer_ctc/exp_"$dataset"_bpe"$bpe" \ - --manifest-dir $datadir/fbank \ - --lang-dir $datadir/lang_bpe_"$bpe" \ - --enable-musan False \ +# ./conformer_ctc/train.py \ +# --num-epochs 60 \ +# --max-duration 300 \ +# --exp-dir ./conformer_ctc/exp_"$dataset"_bpe"$bpe" \ +# --manifest-dir $datadir/fbank \ +# --lang-dir $datadir/lang_bpe_"$bpe" \ +# --enable-musan False \ ./conformer_ctc/decode.py \ - --epoch 60 \ + --epoch 59 \ --avg 10 \ --manifest-dir $datadir/fbank \ --exp-dir ./conformer_ctc/exp_"$dataset"_bpe"$bpe" \ - --max-duration 100 \ - --lang-dir $datadir/lang_bpe_"$bpe" + --max-duration 5 \ + --lang-dir $datadir/lang_bpe_"$bpe" \ + --lm-dir $datadir/"lm" \ + --method $decode_method \ + --num-paths $num_paths \ + \ No newline at end of file