Check both

huggingface · Oct 13, 2023 · 77ff2d0 · 77ff2d0
1 parent d711389
commit 77ff2d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/accelerate/data_loader.py b/src/accelerate/data_loader.py
@@ -455,7 +455,7 @@ def set_epoch(self, epoch: int):
         # In case it is manually passed in, the user can set it to what they like
         if self.iteration != epoch:
             self.iteration = epoch
-        if hasattr(self.batch_sampler.sampler, "set_epoch"):
+        if hasattr(self.batch_sampler, "sampler") and hasattr(self.batch_sampler.sampler, "set_epoch"):
             self.batch_sampler.sampler.set_epoch(epoch)
         elif hasattr(self.dataset, "set_epoch"):
             self.dataset.set_epoch(epoch)