def __init__()

in 01-byoc/code/dataloader.py [0:0]

• 15 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, dataset, batch_size, shuffle=False, validation_split=0.0, num_workers=0, collate_fn=default_collate, pin_memory=True):
        self.validation_split = validation_split
        self.shuffle = shuffle

        self.batch_idx = 0
        self.n_samples = len(dataset)

        self.sampler, self.valid_sampler = self._split_sampler(self.validation_split)

        self.init_kwargs = {
            'dataset': dataset,
            'batch_size': batch_size,
            'shuffle': self.shuffle,
            'collate_fn': collate_fn,
            'num_workers': num_workers,
            'pin_memory': pin_memory
        }
        super().__init__(sampler=self.sampler, **self.init_kwargs)