diff --git a/src/axolotl/utils/callbacks/__init__.py b/src/axolotl/utils/callbacks/__init__.py index 6bf433319..641c9b162 100644 --- 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