Merge pull request #2686 from higorsilvaa/patch-1

updating to evaluation_strategy
UKPLab · May 28, 2024 · 428a994 · 428a994
2 parents 4e5ab7b + bdff7a5
commit 428a994
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/training/sts/training_stsbenchmark.py b/examples/training/sts/training_stsbenchmark.py
@@ -69,7 +69,7 @@
     fp16=True,  # Set to False if you get an error that your GPU can't run on FP16
     bf16=False,  # Set to True if you have a GPU that supports BF16
     # Optional tracking/debugging parameters:
-    eval_strategy="steps",
+    evaluation_strategy="steps",
     eval_steps=100,
     save_strategy="steps",
     save_steps=100,