Fix MpDeviceLoaderWrapper not having attribute batch_sampler (#2242)

* Fix MpDeviceLoaderWrapper not having attribute batch_sampler * fix style
huggingface · Dec 13, 2023 · ad3a5bc · ad3a5bc
1 parent eafcea0
commit ad3a5bc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/accelerate/data_loader.py b/src/accelerate/data_loader.py
@@ -539,6 +539,10 @@ def total_batch_size(self):
         def total_dataset_length(self):
             return self._loader.total_dataset_length
 
+        @property
+        def batch_sampler(self):
+            return self._loader.batch_sampler
+
 
 class DataLoaderDispatcher(DataLoader, DataLoaderStateMixin):
     """