Example #1
0
 def _get_dataloader():
     dataset = RandomDataset(32, 64)
     sampler = RandomSampler(dataset)
     batch_sampler = BatchSampler(sampler, 2, True)
     batch_sampler.set_epoch = Mock()
     return DataLoader(dataset, batch_sampler=batch_sampler)