Input size error with transforms.Lambda(lambda x: Perturbations(x)[0])

[evil-potato]

AIDE/data/datasets.py

Line 122 in 6725b71

image = transform_before(image)

AIDE/data/datasets.py

Line 34 in 6725b71

transform_before = transforms.Compose([

AIDE/data/datasets.py

Line 35 in 6725b71

transforms.ToTensor(),

AIDE/data/datasets.py

Line 36 in 6725b71

transforms.Lambda(lambda x: Perturbations(x)[0])

Hi,
If the training set is GenImage-sdv1.4 and the input image size is not a multiple of 16 (e.g. 600x395), then using transforms.Lambda(lambda x: Perturbations(x)[0]) will result in an error. How did you solve this problem?

Thanks