mod gemma

2025-02-26 13:56:48 +01:00 · 2025-02-26 13:56:48 +01:00 · 87405935eb
parent b9d7a7adc5
commit 87405935eb
1 changed files with 1 additions and 1 deletions
--- a/gemma.py
+++ b/gemma.py
@ -88,7 +88,7 @@ training_args = TrainingArguments(
    learning_rate=1e-5,
    per_device_train_batch_size=2,
    per_device_eval_batch_size=2,
-    num_train_epochs=32,
+    num_train_epochs=16,
    weight_decay=0.01,
    load_best_model_at_end=True,    # Wczytaj najlepszy model na końcu
    metric_for_best_model="loss",   # Kryterium wyboru najlepszego modelu