Method __len__

script/dataloader.py:63–64

(self)

        # self.transform = Compose([Resize(crop_size), ToTensor()])

    def __len__(self):
        return len(self.image_total_files)

    def __getitem__(self, index):
        # Train set