Hello, This is great work!. I am facing an issue with random color augmentation. out = self.color_aug(x) TypeError: 'tuple' object is not callable Could you please take a look?