in data/custom_dataset_data_loader.py [0:0]
def __iter__(self): for i, data in enumerate(self.dataloader): yield data