update detail.

shibing624 · Aug 1, 2023 · adc5597 · adc5597
1 parent 03350db
commit adc5597
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/supervised_finetuning.py b/supervised_finetuning.py
@@ -902,7 +902,9 @@ def preprocess_function(examples):
     # Training
     if training_args.do_train:
         logger.info("*** Train ***")
-        logger.debug(f"Train dataloader example: {next(iter(trainer.get_train_dataloader()))}")
+        sample = next(iter(trainer.get_train_dataloader()))
+        logger.debug(f"Train dataloader example: {sample}")
+        logger.debug(f"Details: \ninput_ids: {list(sample['input_ids'])}, \nlabels: {list(sample['labels'])}")
         checkpoint = None
         if training_args.resume_from_checkpoint is not None:
             checkpoint = training_args.resume_from_checkpoint