Spaces:

ma-xu
/

LIVE

Runtime error

Xu Ma commited on Apr 24, 2022

Commit

505661d

•

1 Parent(s): feef2d1

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -521,7 +521,7 @@ def main_func(target, experiment, cfg_arg):
                 ### added for app
                 if t%10==0:
-                    print(f"debug: {t}, {filename}")
                     return img, None
             x = img.unsqueeze(0).permute(0, 3, 1, 2) # HWC -> NCHW

                 ### added for app
                 if t%10==0:
+                    print(f"debug: {t}, {filename} {img.size()}")
                     return img, None
             x = img.unsqueeze(0).permute(0, 3, 1, 2) # HWC -> NCHW