Fix class name

pull/212/head
Santiago Castro 4 years ago committed by GitHub
parent feaa3abc51
commit 0f75d48a66
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -231,9 +231,9 @@ class MultiEpochsDataLoader(torch.utils.data.DataLoader):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self._DataLoader__initialized = False self._DataLoader__initialized = False
if self.batch_sampler is None: if self.batch_sampler is None:
self.sampler = RepeatSampler(self.sampler) self.sampler = _RepeatSampler(self.sampler)
else: else:
self.batch_sampler = RepeatSampler(self.batch_sampler) self.batch_sampler = _RepeatSampler(self.batch_sampler)
self._DataLoader__initialized = True self._DataLoader__initialized = True
self.iterator = super().__iter__() self.iterator = super().__iter__()

Loading…
Cancel
Save