def __iter__()

in data/custom_dataset_data_loader.py [0:0]


    def __iter__(self):
        for i, data in enumerate(self.dataloader):
            yield data