AlexN commited on
Commit
b0b32a4
1 Parent(s): de5eebc
Files changed (1) hide show
  1. run.sh +7 -7
run.sh CHANGED
@@ -5,11 +5,11 @@ python run_speech_recognition_ctc.py \
5
  --dataset_config_name="fr" \
6
  --output_dir="./" \
7
  --overwrite_output_dir \
8
- --num_train_epochs="5" \
9
  --per_device_train_batch_size="64" \
10
  --per_device_eval_batch_size="64" \
11
  --gradient_accumulation_steps="1" \
12
- --learning_rate="7e-5" \
13
  --warmup_steps="1500" \
14
  --length_column_name="input_length" \
15
  --evaluation_strategy="steps" \
@@ -18,19 +18,19 @@ python run_speech_recognition_ctc.py \
18
  --eval_steps="500" \
19
  --logging_steps="100" \
20
  --layerdrop="0.0" \
21
- --activation_dropout="0.1" \
22
- --save_total_limit="3" \
23
  --freeze_feature_encoder \
24
  --feat_proj_dropout="0.0" \
25
  --mask_time_prob="0.75" \
26
  --mask_time_length="10" \
27
- --mask_feature_prob="0.33" \
28
  --mask_feature_length="10" \
29
  --gradient_checkpointing \
30
  --report_to="wandb" \
31
  --run_name="xls-r-300m-fr" \
32
- --max_eval_samples="4000" \
33
- --max_duration_in_seconds="5" \
34
  --use_auth_token \
35
  --fp16 \
36
  --group_by_length \
 
5
  --dataset_config_name="fr" \
6
  --output_dir="./" \
7
  --overwrite_output_dir \
8
+ --num_train_epochs="2" \
9
  --per_device_train_batch_size="64" \
10
  --per_device_eval_batch_size="64" \
11
  --gradient_accumulation_steps="1" \
12
+ --learning_rate="1e-4" \
13
  --warmup_steps="1500" \
14
  --length_column_name="input_length" \
15
  --evaluation_strategy="steps" \
 
18
  --eval_steps="500" \
19
  --logging_steps="100" \
20
  --layerdrop="0.0" \
21
+ --activation_dropout="0.05" \
22
+ --save_total_limit="2" \
23
  --freeze_feature_encoder \
24
  --feat_proj_dropout="0.0" \
25
  --mask_time_prob="0.75" \
26
  --mask_time_length="10" \
27
+ --mask_feature_prob="0.4" \
28
  --mask_feature_length="10" \
29
  --gradient_checkpointing \
30
  --report_to="wandb" \
31
  --run_name="xls-r-300m-fr" \
32
+ --max_eval_samples="6000" \
33
+ --max_duration_in_seconds="9" \
34
  --use_auth_token \
35
  --fp16 \
36
  --group_by_length \