diff --git a/src/axolotl/core/trainers/base.py b/src/axolotl/core/trainers/base.py index 3427a0b86..627f8e3f8 100644 --- a/src/axolotl/core/trainers/base.py +++ b/src/axolotl/core/trainers/base.py @@ -371,6 +371,11 @@ class AxolotlTrainer( num_items_in_batch=num_items_in_batch, ) + @override + def evaluate(self, *args, **kwargs): + LOG.info("Running evaluation step...") + return super().evaluate(*args, **kwargs) + @staticmethod def orpo_concatenate_inputs(inputs, label_pad_token=-100, pad_token=0, device=None): concatenated_batch = {}