debug

run_continue_nst.sh

@@ -3,8 +3,8 @@
 # Reduce batch size and learning rate if training on smaller GPU
 
 python run_whisper_finetuning.py \
-	--model_name_or_path="../whisper-NST" \
-	--output_dir="../whisper-NST" \
+	--model_name_or_path="openai/whisper-small" \
+	--output_dir="../whisper-NST-cons2e5" \
 	--overwrite_output_dir=True \
 	--language="Norwegian" \
 	--task="transcribe" \
@@ -14,9 +14,10 @@ python run_whisper_finetuning.py \
 	--do_eval=True \
 	--audio_column_name="audio" \
 	--text_column_name="text" \
-	--per_device_train_batch_size=24 \
-    	--per_device_train_batch_size=24 \
+	--per_device_train_batch_size=16 \
+    	--per_device_train_batch_size=16 \
 	--learning_rate=2e-5 \
+	--lr_scheduler_type="constant_with_warmup" \
 	--warmup_steps=500 \
 	--max_steps=10000 \
 	--gradient_checkpointing=True \