siamese_model.py
class SiameseDataset(Dataset):
    def __init__(self, data, transform=None):
        ...
        
    def __len__(self):
        return len(self.data)

Dataset

class