Commit
·
b324d03
1
Parent(s):
0829d49
try to increase batch size
Browse files
run.sh
CHANGED
|
@@ -7,8 +7,8 @@ python run_speech_recognition_ctc.py \
|
|
| 7 |
--output_dir="./" \
|
| 8 |
--overwrite_output_dir \
|
| 9 |
--num_train_epochs="3" \
|
| 10 |
-
--per_device_train_batch_size="
|
| 11 |
-
--per_device_eval_batch_size="
|
| 12 |
--gradient_accumulation_steps="4" \
|
| 13 |
--learning_rate="7.5e-5" \
|
| 14 |
--warmup_ratio="0.02" \
|
|
@@ -23,7 +23,7 @@ python run_speech_recognition_ctc.py \
|
|
| 23 |
--dataset_seed="42" \
|
| 24 |
--layerdrop="0.0" \
|
| 25 |
--activation_dropout="0.1" \
|
| 26 |
-
--save_total_limit="
|
| 27 |
--freeze_feature_encoder \
|
| 28 |
--feat_proj_dropout="0.0" \
|
| 29 |
--mask_time_prob="0.75" \
|
|
|
|
| 7 |
--output_dir="./" \
|
| 8 |
--overwrite_output_dir \
|
| 9 |
--num_train_epochs="3" \
|
| 10 |
+
--per_device_train_batch_size="32" \
|
| 11 |
+
--per_device_eval_batch_size="32" \
|
| 12 |
--gradient_accumulation_steps="4" \
|
| 13 |
--learning_rate="7.5e-5" \
|
| 14 |
--warmup_ratio="0.02" \
|
|
|
|
| 23 |
--dataset_seed="42" \
|
| 24 |
--layerdrop="0.0" \
|
| 25 |
--activation_dropout="0.1" \
|
| 26 |
+
--save_total_limit="2" \
|
| 27 |
--freeze_feature_encoder \
|
| 28 |
--feat_proj_dropout="0.0" \
|
| 29 |
--mask_time_prob="0.75" \
|