.

huggingface · May 6, 2024 · aac6e7b · aac6e7b
1 parent 8864d94
commit aac6e7b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/custom-dataloader/README.md b/examples/custom-dataloader/README.md
@@ -8,7 +8,7 @@ https://github.com/huggingface/nanotron/blob/2e21db0db46a40bedbd03714616dd0ae4ea
 And to test it out, you should fix your config to have: (example: [config_custom_dl.yaml](config_custom_dl.yaml))
 ```yaml
 - data:
-    dataset: null
+    dataset: null # Custom dataloader will be used
     num_loading_workers: 1
     seed: 42
   name: Stable Training Stage

diff --git a/examples/custom-dataloader/config_custom_dl.yaml b/examples/custom-dataloader/config_custom_dl.yaml
@@ -6,7 +6,7 @@ checkpoints:
   save_initial_state: false
 data_stages:
 - data:
-    dataset: null
+    dataset: null # Custom dataloader will be used
     num_loading_workers: 1
     seed: 42
   name: Stable Training Stage