From bdff7a5e534165d9e91a3dd36707afb2c02168a9 Mon Sep 17 00:00:00 2001 From: higorsilvah <74798202+higorsilvaa@users.noreply.github.com> Date: Tue, 28 May 2024 11:09:31 -0300 Subject: [PATCH] updating to evaluation_strategy There were an eval_strategy parameter that does not exist. --- examples/training/sts/training_stsbenchmark.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/training/sts/training_stsbenchmark.py b/examples/training/sts/training_stsbenchmark.py index c61648494..4796cfdca 100644 --- 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,