Example #1
0
 def test_dataloader(self):
     return [torch.utils.data.DataLoader(RandomDataset(32, 64)) for _ in range(num_dataloaders)]
 def test_dataloader(self):
     return torch.utils.data.DataLoader(RandomDataset(32, 64))
Example #3
0
 def train_dataloader(self):
     return torch.utils.data.DataLoader(RandomDataset(32, 64),
                                        collate_fn=collate_fn)