patrickvonplaten commited on
Commit
d8b053b
1 Parent(s): 6ebc6d9

Update run.sh

Browse files
Files changed (1) hide show
  1. run.sh +4 -4
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/wavlbasege" \
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
- --per_device_train_batch_size4"2" \
15
- --per_device_eval_batch_size4"2" \
16
- --gradient_accumulation_steps1"2" \
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" \