def _get_data_loader()

in torchrecipes/vision/data/modules/torchvision_data_module.py [0:0]

• 13 lines of code
• 2 McCabe index (conditional complexity)

    def _get_data_loader(self, dataset: VisionDataset, phase: str) -> DataLoader:
        if phase == "train":
            sampler = RandomSampler(dataset)
        else:
            sampler = SequentialSampler(dataset)
        return DataLoader(
            dataset,
            sampler=sampler,
            batch_size=self.batch_size,
            num_workers=self.num_workers,
            drop_last=self.drop_last,
            pin_memory=self.pin_memory,
        )