Commit
·
d8b053b
1
Parent(s):
6ebc6d9
Update run.sh
Browse files
run.sh
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
python -m torch.distributed.launch \
|
| 3 |
--nproc_per_node 8 run_speech_recognition_ctc.py \
|
| 4 |
--dataset_name="librispeech_asr" \
|
| 5 |
-
--model_name_or_path="microsoft/
|
| 6 |
--dataset_config_name="clean" \
|
| 7 |
--train_split_name="train.100" \
|
| 8 |
--eval_split_name="validation" \
|
|
@@ -11,9 +11,9 @@ python -m torch.distributed.launch \
|
|
| 11 |
--length_column_name="input_length" \
|
| 12 |
--overwrite_output_dir \
|
| 13 |
--num_train_epochs="3" \
|
| 14 |
-
--
|
| 15 |
-
--
|
| 16 |
-
--
|
| 17 |
--learning_rate="3e-4" \
|
| 18 |
--warmup_steps="500" \
|
| 19 |
--evaluation_strategy="steps" \
|
|
|
|
| 2 |
python -m torch.distributed.launch \
|
| 3 |
--nproc_per_node 8 run_speech_recognition_ctc.py \
|
| 4 |
--dataset_name="librispeech_asr" \
|
| 5 |
+
--model_name_or_path="microsoft/wavlm-base" \
|
| 6 |
--dataset_config_name="clean" \
|
| 7 |
--train_split_name="train.100" \
|
| 8 |
--eval_split_name="validation" \
|
|
|
|
| 11 |
--length_column_name="input_length" \
|
| 12 |
--overwrite_output_dir \
|
| 13 |
--num_train_epochs="3" \
|
| 14 |
+
--per_device_train_batch_size="4" \
|
| 15 |
+
--per_device_eval_batch_size="4" \
|
| 16 |
+
--gradient_accumulation_steps="1" \
|
| 17 |
--learning_rate="3e-4" \
|
| 18 |
--warmup_steps="500" \
|
| 19 |
--evaluation_strategy="steps" \
|