sanchit-gandhi HF staff commited on
Commit
09f6df2
1 Parent(s): efe4c6f

Correct scripts

Browse files
Files changed (2) hide show
  1. README.md +6 -4
  2. run_spgispeech.sh +6 -4
README.md CHANGED
@@ -21,12 +21,14 @@ python run_flax_speech_recognition_seq2seq.py \
21
  --learning_rate="1e-4" \
22
  --warmup_steps="500" \
23
  --logging_steps="25" \
24
- --max_steps="50000" \
25
  --eval_steps="10000" \
26
  --save_steps="10000" \
27
- --generation_max_length="200" \
28
- --generation_num_beams="5" \
29
- --generation_length_penalty="1.2" \
 
 
30
  --overwrite_output_dir \
31
  --gradient_checkpointing \
32
  --freeze_feature_encoder \
 
21
  --learning_rate="1e-4" \
22
  --warmup_steps="500" \
23
  --logging_steps="25" \
24
+ --max_steps="50001" \
25
  --eval_steps="10000" \
26
  --save_steps="10000" \
27
+ --generation_max_length="40" \
28
+ --generation_num_beams="1" \
29
+ --final_generation_max_length="225" \
30
+ --final_generation_num_beams="14" \
31
+ --generation_length_penalty="1.6" \
32
  --overwrite_output_dir \
33
  --gradient_checkpointing \
34
  --freeze_feature_encoder \
run_spgispeech.sh CHANGED
@@ -11,12 +11,14 @@ python run_flax_speech_recognition_seq2seq.py \
11
  --learning_rate="1e-4" \
12
  --warmup_steps="500" \
13
  --logging_steps="25" \
14
- --max_steps="50000" \
15
  --eval_steps="10000" \
16
  --save_steps="10000" \
17
- --generation_max_length="200" \
18
- --generation_num_beams="5" \
19
- --generation_length_penalty="1.2" \
 
 
20
  --overwrite_output_dir \
21
  --gradient_checkpointing \
22
  --freeze_feature_encoder \
 
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 \