changes

3 years ago · 0e8901749e
1 changed files with 2 additions and 1 deletions
--- a/animegan.py
+++ b/animegan.py
@ -30,5 +30,6 @@ class Animegan(Operator):
    @arg(1, to_image_color('RGB'))
    def __call__(self, image):
        img = self.tfms(image).unsqueeze(0)
+        print(img.shape)
        styled_image = self.model(img)
        return Image(styled_image, 'RGB')