whisper-flaxtest / run_streaming.sh
versae's picture
Clean up files
eac79a2
python run_flax_speech_recognition_seq2seq_streaming.py \
--model_name_or_path openai/whisper-tiny.en \
--dataset_name mozilla-foundation/common_voice_11_0 \
--dataset_config nn-NO \
--text_column_name sentence \
--train_split_name test\
--eval_split_name test\
--output_dir whisper-tiny-ft-dummy\
--overwrite_output_dir\
--max_eval_samples=16\
--warmup_steps=8 \
--do_train \
--do_eval \
--num_train_steps 30 \
--eval_steps 10 \
--learning_rate=2e-4 \
--per_device_train_batch_size=2 \
--per_device_eval_batch_size=2 \
--predict_with_generate \
--streaming=True