evaluation_strategy was fully deprecated in recent release (#2169) [s…

…kip ci]
axolotl-ai-cloud · Dec 12, 2024 · 7b97876 · 7b97876
1 parent 214022b
commit 7b97876
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/axolotl/utils/callbacks/__init__.py b/src/axolotl/utils/callbacks/__init__.py
@@ -66,10 +66,7 @@ def on_step_end(
         control: TrainerControl,
         **kwargs,
     ):
-        if (
-            args.evaluation_strategy == IntervalStrategy.STEPS
-            and state.global_step == 1
-        ):
+        if args.eval_strategy == IntervalStrategy.STEPS and state.global_step == 1:
             control.should_evaluate = True
         return control