marinone94 commited on
Commit
67e34d0
1 Parent(s): 688bd6b

define num steps

Browse files
Files changed (1) hide show
  1. run.sh +1 -1
run.sh CHANGED
@@ -9,7 +9,7 @@ python $1run_speech_recognition_seq2seq_streaming.py \
9
  --language_eval="sv" \
10
  --eval_split_name="test" \
11
  --model_index_name="Whisper Medium Nordic" \
12
- --num_train_epochs="1" \
13
  --output_dir="./" \
14
  --per_device_train_batch_size="32" \
15
  --per_device_eval_batch_size="16" \
 
9
  --language_eval="sv" \
10
  --eval_split_name="test" \
11
  --model_index_name="Whisper Medium Nordic" \
12
+ --max_steps="10000" \
13
  --output_dir="./" \
14
  --per_device_train_batch_size="32" \
15
  --per_device_eval_batch_size="16" \