Views
No views yet
1#!/usr/bin/env bash
2python run_flax_speech_recognition_seq2seq.py \
3 --dataset_name="esb/datasets" \
4 --model_name_or_path="esb/wav2vec2-aed-pretrained" \
5 --dataset_config_name="spgispeech" \
6 --output_dir="./" \
7 --wandb_name="wav2vec2-aed-spgispeech" \
8 --wandb_project="wav2vec2-aed" \
9 --per_device_train_batch_size="8" \
10 --per_device_eval_batch_size="2" \
11 --learning_rate="1e-4" \
12 --warmup_steps="500" \
13 --logging_steps="25" \
14 --max_steps="50001" \
15 --eval_steps="10000" \
16 --save_steps="10000" \
17 --generation_max_length="40" \
18 --generation_num_beams="1" \
19 --final_generation_max_length="225" \
20 --final_generation_num_beams="14" \
21 --generation_length_penalty="1.6" \
22 --overwrite_output_dir \
23 --gradient_checkpointing \
24 --freeze_feature_encoder \
25 --predict_with_generate \
26 --do_eval \
27 --do_train \
28 --do_predict \
29 --push_to_hub \
30 --use_auth_token