log before eval step (#3148) [skip-ci]

This commit is contained in:
salman
2025-09-11 11:19:30 +01:00
committed by GitHub
parent 1b53c49e1a
commit 9406c0c488

View File

@@ -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 = {}