Spaces:
Runtime error
Runtime error
Xu Ma
commited on
Commit
•
e10c15b
1
Parent(s):
9be2593
Update main.py
Browse files
main.py
CHANGED
|
@@ -370,6 +370,7 @@ def main_func(target, experiment, cfg_arg):
|
|
| 370 |
pydiffvg.set_use_gpu(torch.cuda.is_available())
|
| 371 |
device = pydiffvg.get_device()
|
| 372 |
|
|
|
|
| 373 |
gt = np.array(PIL.Image.open(cfg.target))
|
| 374 |
print(f"Input image shape is: {gt.shape}")
|
| 375 |
if len(gt.shape) == 2:
|
|
|
|
| 370 |
pydiffvg.set_use_gpu(torch.cuda.is_available())
|
| 371 |
device = pydiffvg.get_device()
|
| 372 |
|
| 373 |
+
print(f"debug: {cfg.target}")
|
| 374 |
gt = np.array(PIL.Image.open(cfg.target))
|
| 375 |
print(f"Input image shape is: {gt.shape}")
|
| 376 |
if len(gt.shape) == 2:
|