fix docstring typo (#2072)

huggingface · Oct 24, 2023 · f322987 · f322987
1 parent 7843286
commit f322987
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/accelerate/data_loader.py b/src/accelerate/data_loader.py
@@ -406,7 +406,7 @@ class DataLoaderShard(DataLoader, DataLoaderStateMixin):
             - `"generator"`: an optional `torch.Generator`
         synchronized_generator (`torch.Generator`, *optional*):
             A random number generator to keep synchronized across processes.
-        split_batches (`int`, *optional*, defaults to 0):
+        skip_batches (`int`, *optional*, defaults to 0):
             The number of batches to skip at the beginning.
         kwargs:
             All other keyword arguments to pass to the regular `DataLoader` initialization.