From b2e429da3e2afda6185fb9c1d27369b83544df33 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 26 Dec 2024 18:21:30 -0800 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index c9ca3eb1e5..5fa6b5de52 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1009,6 +1009,7 @@ def _unsloth_get_batch_samples(self, epoch_iterator, num_batches): except StopIteration: break + print("NUM_ITMES = ", num_items_in_batch) if len(batch_samples) > 0 and "labels" in batch_samples[0]: try: num_items_in_batch = sum([(batch["labels"].ne(-100)).sum() for batch in batch_samples])