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