edits

app.py

@@ -49,7 +49,7 @@ class ImgDataset(data.Dataset):
     def __getitem__(self, index):
         img = self.images[index]
         img.thumbnail((self.imsize, self.imsize), Image.ANTIALIAS)
-        print(img.size)
+        print('after imresize:', img.size)
         return  self.transform(img)