Fix class name

4 years ago · 0f75d48a66
parent feaa3abc51
commit 0f75d48a66
1 changed files with 2 additions and 2 deletions
--- a/timm/data/loader.py
+++ b/timm/data/loader.py
@ -231,9 +231,9 @@ class MultiEpochsDataLoader(torch.utils.data.DataLoader):
        super().__init__(*args, **kwargs)
        self._DataLoader__initialized = False
        if self.batch_sampler is None:
-            self.sampler = RepeatSampler(self.sampler)
+            self.sampler = _RepeatSampler(self.sampler)
        else:
-            self.batch_sampler = RepeatSampler(self.batch_sampler)
+            self.batch_sampler = _RepeatSampler(self.batch_sampler)
        self._DataLoader__initialized = True
        self.iterator = super().__iter__()