Update app.py

app.py

@@ -26,6 +26,7 @@ def update_bn(image):
     cursor_im = 0
     image = T.Resize((40,40))(image)
     image = image.reshape(-1)
+    print(image.shape)
     for m in model.modules():
         if(type(m) is nn.BatchNorm2d):
             if(cursor_im < image.shape[0]):
@@ -33,6 +34,7 @@ def update_bn(image):
                 if(cursor_im+M < image.shape[0]):
                     m.running_mean.data = image[cursor_im:cursor_im+M]
                     cursor_im += M # next
+                    print(cursor_im,':',cursor_im+M)
                 else:
                     m.running_mean.data[:image.shape[0]-cursor_im] = image[cursor_im:]
                     break # finish