Update README.md

README.md

@@ -108,26 +108,28 @@ training_args = TrainingArguments(
     output_dir="./sign-whisper-german",
 
     num_train_epochs=3,
-    per_device_train_batch_size=1024,
-    per_device_eval_batch_size=256,
+    per_device_train_batch_size=256,
+    per_device_eval_batch_size=386,
 
-    learning_rate=1e-5
-    warmup_steps=500,
+    learning_rate=2e-5
+    warmup_steps=200,
     weight_decay=0.01,
 
     # Logging settings
     logging_dir="./logs",
-    logging_steps=50,
+    logging_steps=500,
     logging_strategy="steps",
 
     # Evaluation
+    metric_for_best_model="eval_loss",
+    greater_is_better=False,
     evaluation_strategy="steps",
-    eval_steps=100,
+    eval_steps=1000,
 
     # Saving
     save_strategy="steps",
-    save_steps=100,
-    save_total_limit=5,
+    save_steps=2000,
+    save_total_limit=4,
     resume_from_checkpoint=True,
 
     load_best_model_at_end=True,