From 9406c0c488277ef9d7152568b9fda50600c4221e Mon Sep 17 00:00:00 2001 From: salman Date: Thu, 11 Sep 2025 11:19:30 +0100 Subject: [PATCH] log before eval step (#3148) [skip-ci] --- src/axolotl/core/trainers/base.py | 5 +++++ 1 file changed, 5 insertions(+) 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 = {}