pere commited on
Commit
9e8fbe1
1 Parent(s): 3d86552
Files changed (1) hide show
  1. run.sh +3 -3
run.sh CHANGED
@@ -7,9 +7,9 @@ python run_speech_recognition_seq2seq_streaming.py \
7
  --model_index_name="Whisper Tiny Norwegian Bokmål" \
8
  --max_steps="100000" \
9
  --output_dir="./" \
10
- --per_device_train_batch_size="64" \
11
- --per_device_eval_batch_size="32" \
12
- --gradient_accumulation_step="2" \
13
  --logging_steps="50" \
14
  --learning_rate="1e-5" \
15
  --lr_scheduler_type="constant_with_warmup" \
 
7
  --model_index_name="Whisper Tiny Norwegian Bokmål" \
8
  --max_steps="100000" \
9
  --output_dir="./" \
10
+ --per_device_train_batch_size="128" \
11
+ --per_device_eval_batch_size="64" \
12
+ --gradient_accumulation_step="1" \
13
  --logging_steps="50" \
14
  --learning_rate="1e-5" \
15
  --lr_scheduler_type="constant_with_warmup" \