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

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

View File

@ -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:

View File

@ -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 \