I am following the Data Loading Tutorial as I am facing a face key point detector problem, but I have an error when executing its code.
If I create the Dataset FaceLandmarksDataset(Dataset) like proposed:
class FaceLandmarksDataset(Dataset):
    ''' Face Pose Detection Dataset '''
    
    def __init__(self, csv, root, transform=None):
        
        self.land_file = pd.read_csv(csv)
        self.root_dir  = root
        self.transform = transform
        
    def __len__(self):
        
        return len(self.land_file)
    
    def __getitem__(self, idx):
        
        img_name = os.path.join(self.root_dir, self.land_file.iloc[idx,0])
        image = io.imread(img_name)
        landmarks = self.land_file.iloc[idx,1:].values.astype('float').reshape(-1, 2)
        sample = {'image': image, 'landmarks': landmarks}
        
        if self.transform: 
            sample = self.transform(sample)
        
        return sample

I get the following error:
TypeError: module.__init__() takes at most 2 arguments (3 given)
Why can’t I customize the init method?
Thanks in advance,
Pablo