mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 18:12:19 +00:00
39 lines
972 B
Bash
Executable File
39 lines
972 B
Bash
Executable File
#!/bin/bash
|
|
export CUDA_VISIBLE_DEVICES="0"
|
|
|
|
set -e
|
|
dataset='hi-en'
|
|
datadir=data_"$dataset"
|
|
bpe=400
|
|
# decode_methods="attention-decoder 1best nbest nbest-rescoring ctc-decoding whole-lattice-rescoring"
|
|
decode_methods="nbest nbest-rescoring whole-lattice-rescoring"
|
|
|
|
num_paths=10
|
|
max_duration=5
|
|
|
|
# ./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 \
|
|
|
|
for decode_method in $decode_methods;
|
|
do
|
|
|
|
./conformer_ctc/decode.py \
|
|
--epoch 59 \
|
|
--avg 10 \
|
|
--manifest-dir $datadir/fbank \
|
|
--exp-dir ./conformer_ctc/exp_"$dataset"_bpe"$bpe" \
|
|
--max-duration $max_duration \
|
|
--lang-dir $datadir/lang_bpe_"$bpe" \
|
|
--lm-dir $datadir/"lm" \
|
|
--method $decode_method \
|
|
--num-paths $num_paths \
|
|
|
|
done
|
|
exit
|
|
|
|
|