Merge pull request #140 from huggingface/xrsrke/fix_print_data_stages…

…_break_in_brrr

[Bug] Remove printing of HF dataset in data stages

src/nanotron/trainer.py

@@ -335,7 +335,7 @@ def clear_dataloader_from_memory(dataloader: DataLoader, stage_name: str):
                         clear_dataloader_from_memory(prev_dataloader, stage_name=stage.name)
 
                 log_rank(
-                    f"[Training Stage: {stage.name}] Switching to a new dataset {stage.data.dataset.hf_dataset_or_datasets}",
+                    f"[Training Stage: {stage.name}] Switching to a new dataset",
                     logger=logger,
                     level=logging.INFO,
                     rank=0,