Spaces:

morinop
/

hellominori

Sleeping

morinop commited on Mar 3, 2023

Commit

a800e1f

•

1 Parent(s): 7c4c927

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ def print_bn():
 def update_bn(image):
     cursor_im = 0
-    image = image.view(-1)
     for m in model.modules():
         if(type(m) is nn.BatchNorm2d):
             if(cursor_im < image.shape[0]):

 def update_bn(image):
     cursor_im = 0
+    image = image.reshape(-1)
     for m in model.modules():
         if(type(m) is nn.BatchNorm2d):
             if(cursor_im < image.shape[0]):