esc-bench
/

wav2vec2-aed-tedlium

Automatic Speech Recognition

speech-encoder-decoder

Inference Endpoints

Model card Files Files and versions Community

sanchit-gandhi HF staff commited on Oct 4, 2022

Commit

8033e67

•

1 Parent(s): dfb2466

Correct scripts

Files changed (2) hide show

README.md +6 -4
run_tedlium.sh +6 -4

README.md CHANGED Viewed

@@ -21,12 +21,14 @@ python run_flax_speech_recognition_seq2seq.py \
         --learning_rate="1e-4" \
         --warmup_steps="500" \
         --logging_steps="25" \
-        --max_steps="50000" \
         --eval_steps="10000" \
         --save_steps="10000" \
-        --generation_max_length="200" \
-        --generation_num_beams="5" \
-        --generation_length_penalty="1.2" \
         --hidden_dropout="0.2" \
         --activation_dropout="0.2" \
         --feat_proj_dropout="0.2" \

         --learning_rate="1e-4" \
         --warmup_steps="500" \
         --logging_steps="25" \
+        --max_steps="50001" \
         --eval_steps="10000" \
         --save_steps="10000" \
+        --generation_max_length="40" \
+        --generation_num_beams="1" \
+        --final_generation_max_length="250" \
+        --final_generation_num_beams="12" \
+        --generation_length_penalty="1.5" \
         --hidden_dropout="0.2" \
         --activation_dropout="0.2" \
         --feat_proj_dropout="0.2" \

run_tedlium.sh CHANGED Viewed

@@ -11,12 +11,14 @@ python run_flax_speech_recognition_seq2seq.py \
         --learning_rate="1e-4" \
         --warmup_steps="500" \
         --logging_steps="25" \
-        --max_steps="50000" \
         --eval_steps="10000" \
         --save_steps="10000" \
-        --generation_max_length="200" \
-        --generation_num_beams="5" \
-        --generation_length_penalty="1.2" \
         --hidden_dropout="0.2" \
         --activation_dropout="0.2" \
         --feat_proj_dropout="0.2" \

         --learning_rate="1e-4" \
         --warmup_steps="500" \
         --logging_steps="25" \
+        --max_steps="50001" \
         --eval_steps="10000" \
         --save_steps="10000" \
+        --generation_max_length="40" \
+        --generation_num_beams="1" \
+        --final_generation_max_length="250" \
+        --final_generation_num_beams="12" \
+        --generation_length_penalty="1.5" \
         --hidden_dropout="0.2" \
         --activation_dropout="0.2" \
         --feat_proj_dropout="0.2" \