====================================================
Transforms
====================================================
def get_transforms(*, data):
if data == 'train':
    return A.Compose([
        ToTensorV2(),
    ])

elif data == 'valid':
    return A.Compose([
        ToTensorV2(),
    ])


Is there a solution to this that would use this? I’m really sorry for the late reply.