Spaces:

ehabhussien
/

marbertga-training

Running

ehabhussien commited on 16 days ago

Commit

df36b33

verified ·

1 Parent(s): dae8156

fix: eval_strategy (was evaluation_strategy) — correct Space path this time

Files changed (1) hide show

hf_pipeline/trainer.py CHANGED Viewed

@@ -251,7 +251,7 @@ def fine_tune_marbert(
         learning_rate=lr,
         warmup_ratio=warmup_ratio,
         weight_decay=weight_decay,
-        evaluation_strategy="epoch",
         save_strategy="epoch",
         load_best_model_at_end=True,
         metric_for_best_model="f1",

         learning_rate=lr,
         warmup_ratio=warmup_ratio,
         weight_decay=weight_decay,
+        eval_strategy="epoch",
         save_strategy="epoch",
         load_best_model_at_end=True,
         metric_for_best_model="f1",