patrickvonplaten
/

wavlm-libri-clean-100h-base

Automatic Speech Recognition

librispeech_asr

Generated from Trainer

wavlm_libri_finetune

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

patrickvonplaten commited on Dec 20, 2021

Commit

d8b053b

•

1 Parent(s): 6ebc6d9

Update run.sh

Files changed (1) hide show

run.sh +4 -4

run.sh CHANGED Viewed

@@ -2,7 +2,7 @@
 python -m torch.distributed.launch \
  	--nproc_per_node 8 run_speech_recognition_ctc.py \
 	--dataset_name="librispeech_asr" \
-	--model_name_or_path="microsoft/wavlbasege" \
 	--dataset_config_name="clean" \
 	--train_split_name="train.100" \
 	--eval_split_name="validation" \
@@ -11,9 +11,9 @@ python -m torch.distributed.launch \
 	--length_column_name="input_length" \
 	--overwrite_output_dir \
 	--num_train_epochs="3" \
-	--per_device_train_batch_size4"2" \
-	--per_device_eval_batch_size4"2" \
-	--gradient_accumulation_steps1"2" \
 	--learning_rate="3e-4" \
 	--warmup_steps="500" \
 	--evaluation_strategy="steps" \

 python -m torch.distributed.launch \
  	--nproc_per_node 8 run_speech_recognition_ctc.py \
 	--dataset_name="librispeech_asr" \
+	--model_name_or_path="microsoft/wavlm-base" \
 	--dataset_config_name="clean" \
 	--train_split_name="train.100" \
 	--eval_split_name="validation" \
 	--length_column_name="input_length" \
 	--overwrite_output_dir \
 	--num_train_epochs="3" \
+	--per_device_train_batch_size="4" \
+	--per_device_eval_batch_size="4" \
+	--gradient_accumulation_steps="1" \
 	--learning_rate="3e-4" \
 	--warmup_steps="500" \
 	--evaluation_strategy="steps" \