775f58f3ff3802d484cd74564ce1496ddfbe11fc,kornia/augmentation/augmentation.py,RandomVerticalFlip,forward,#RandomVerticalFlip#Any#Any#,144

Before Change

    def forward(self, input: UnionType, params: Optional[Dict[str, torch.Tensor]] = None) -> UnionType:  // type: ignore
        if params is None:
            batch_size = self.infer_batch_size(input)
            params = self.get_params(batch_size, self.p)
        return super().forward(input, params)


class ColorJitter(AugmentationBase):

After Change

    def forward(self, input: UnionType, params: Optional[Dict[str, torch.Tensor]] = None) -> UnionType:  // type: ignore
        if params is None:
            batch_size = self.infer_batch_size(input)
            self.set_params(batch_size, self.p)
        return super().forward(input, self.params)


class ColorJitter(AugmentationBase):

Instances