Spaces:
Runtime error
Runtime error
Xu Ma
commited on
Commit
·
a85a5c8
1
Parent(s):
99293cd
Update main.py
Browse files
main.py
CHANGED
@@ -521,6 +521,7 @@ def main_func(target, experiment, cfg_arg):
|
|
521 |
|
522 |
### added for app
|
523 |
if t%10==0:
|
|
|
524 |
return filename, None
|
525 |
|
526 |
x = img.unsqueeze(0).permute(0, 3, 1, 2) # HWC -> NCHW
|
|
|
521 |
|
522 |
### added for app
|
523 |
if t%10==0:
|
524 |
+
print(f"debug: {t}, {filename}")
|
525 |
return filename, None
|
526 |
|
527 |
x = img.unsqueeze(0).permute(0, 3, 1, 2) # HWC -> NCHW
|