File size: 1,237 Bytes
4095301 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | #!/usr/bin/env bash
root=~/distil-whisper
cd $root
source path.sh
cd $root/utils
CUDA_VISIBLE_DEVICES=1
# model_id=/root/distil-whisper/corpus/output/baseline
# model_id=/root/distil-whisper/corpus/output/hallucination/6gram_5repeat_detector
# model_id=/root/distil-whisper/corpus/output/hallucination/whisper-base_threshold0.4
# model_id=/root/distil-whisper/corpus/output/hallucination/whisper-base_threshold0.6
# model_id=/root/distil-whisper/corpus/output/hallucination/whisper-base_threshold0.4-phonemized
# model_id=/root/distil-whisper/corpus/output/hallucination/whisper-base_threshold0.6-phonemized
# model_id=/root/distil-whisper/corpus/output/hallucination/6gram-5repeat_base-per0.4_allow-empty
# model_id=/root/distil-whisper/corpus/output/hallucination/6gram-5repeat_base-per0.6_allow-empty
# model_id=openai/whisper-large-v2
# model_id=openai/whisper-medium
model_id=openai/whisper-small
testing_data_root=/root/distil-whisper/corpus/ntu_cool/data/test/for_longform
for lang in zh; do
python longform_eval.py \
--normalize \
--model_id $model_id \
--testing_data_root $testing_data_root \
--device_id $CUDA_VISIBLE_DEVICES \
--language $lang
# --mix_lang_emb
done
|