| python run_flax_speech_recognition_seq2seq.py \ | |
| --model_name_or_path openai/whisper-small.en \ | |
| --dataset_name mozilla-foundation/common_voice_11_0 \ | |
| --dataset_config es \ | |
| --text_column_name sentence \ | |
| --train_split_name test\ | |
| --eval_split_name test\ | |
| --output_dir whisper-small-flaxtest \ | |
| --overwrite_output_dir \ | |
| --num_train_epochs=2 \ | |
| --max_train_samples 64 \ | |
| --max_eval_samples 10 \ | |
| --warmup_steps=8 \ | |
| --do_train \ | |
| --do_eval \ | |
| --learning_rate=2e-4 \ | |
| --per_device_train_batch_size=2 \ | |
| --per_device_eval_batch_size=4 \ | |
| --predict_with_generate | |