danielizham commited on
Commit
f54a820
1 Parent(s): d07d754

Update run.sh

Browse files
Files changed (1) hide show
  1. run.sh +2 -2
run.sh CHANGED
@@ -6,14 +6,14 @@ python run_speech_recognition_seq2seq_streaming.py \
6
  --train_split_name="train+validation" \
7
  --eval_split_name="test" \
8
  --model_index_name="Whisper Small Arabic" \
9
- --num_train_epochs="30" \
10
  --output_dir="./" \
11
  --per_device_train_batch_size="32" \
12
  --gradient_accumulation_steps="8" \
13
  --per_device_eval_batch_size="32" \
14
  --logging_steps="25" \
15
  --learning_rate="1.75e-5" \
16
- --warmup_steps="4000" \
17
  --evaluation_strategy="steps" \
18
  --eval_steps="1000" \
19
  --save_strategy="steps" \
 
6
  --train_split_name="train+validation" \
7
  --eval_split_name="test" \
8
  --model_index_name="Whisper Small Arabic" \
9
+ --max_steps="25000" \
10
  --output_dir="./" \
11
  --per_device_train_batch_size="32" \
12
  --gradient_accumulation_steps="8" \
13
  --per_device_eval_batch_size="32" \
14
  --logging_steps="25" \
15
  --learning_rate="1.75e-5" \
16
+ --warmup_steps="5000" \
17
  --evaluation_strategy="steps" \
18
  --eval_steps="1000" \
19
  --save_strategy="steps" \