cleverk1ng

I think this function should be useful. ``` def joint_transform(img, mask): img = transforms.ToTensor()(img) mask = transforms.ToTensor()(mask) both_images = torch.cat((img, mask), 0) # Apply the transformations to both images simultaneously:...