diff --git a/x2paddle/project_convertor/pytorch/torch2paddle/vision_transforms.py b/x2paddle/project_convertor/pytorch/torch2paddle/vision_transforms.py index f7f283cd56ff01b63646d591743da9ab0111937d..f7da97fcef14199ec50e7982119009234c5ea6d7 100644 --- a/x2paddle/project_convertor/pytorch/torch2paddle/vision_transforms.py +++ b/x2paddle/project_convertor/pytorch/torch2paddle/vision_transforms.py @@ -23,7 +23,7 @@ from paddle.vision.transforms import functional as F class ToPILImage(BaseTransform): def __init__(self, mode=None, keys=None): - super(ToTensor, self).__init__(keys) + super(ToPILImage, self).__init__(keys) self.data_format = data_format def _apply_image(self, pic):