Skip to content
Snippets Groups Projects
Commit 7d08bcbb authored by Jeppe Gade's avatar Jeppe Gade
Browse files

Deleted legacy code

parent 52e10897
No related branches found
No related tags found
No related merge requests found
......@@ -60,36 +60,4 @@ for epoch in range(2):
print(f'[{epoch + 1}, {i + 1:5d}] loss: {sum(running_loss) / 2000:.3f}')
running_loss = 0.0
print('Finished Training')
"""
class MultiEpochsDataLoader(torch.utils.data.DataLoader):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._DataLoader__initialized = False
self.batch_sampler = _RepeatSampler(self.batch_sampler)
self._DataLoader__initialized = True
self.iterator = super().__iter__()
def __len__(self):
return len(self.batch_sampler.sampler)
def __iter__(self):
for i in range(len(self)):
yield next(self.iterator)
class _RepeatSampler(object):
Sampler that repeats forever.
Args:
sampler (Sampler)
def __init__(self, sampler):
self.sampler = sampler
def __iter__(self):
while True:
yield from iter(self.sampler)
"""
\ No newline at end of file
print('Finished Training')
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment