Xu Ma commited on
Commit
a85a5c8
·
1 Parent(s): 99293cd

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
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