evaluation_strategy was fully deprecated in recent release (#2169) [skip ci]

2024-12-11 20:14:24 -05:00
parent d009ead101
commit 78a4aa86d6
1 changed files with 1 additions and 4 deletions
--- a/src/axolotl/utils/callbacks/init.py
+++ b/src/axolotl/utils/callbacks/init.py
@@ -66,10 +66,7 @@ class EvalFirstStepCallback(
        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