fix ToPILImage init (#690)

60ef0f0a · littletomatodonkey · GitHub · bf7fd504 · 60ef0f0a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

x2paddle/project_convertor/pytorch/torch2paddle/vision_transforms.py ...oject_convertor/pytorch/torch2paddle/vision_transforms.py +1 -1

未找到文件。
--- 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):