mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix class name

Browse Source
This commit is contained in:
Santiago Castro 2020-08-07 16:26:00 -04:00 committed by GitHub
parent feaa3abc51
commit 0f75d48a66
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/data/loader.py
View File

@ -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__()