Spaces:
Runtime error
Runtime error
Mehdi Cherti
commited on
Commit
•
9916e2b
1
Parent(s):
d8a4058
update
Browse files- test_ddgan.py +2 -1
test_ddgan.py
CHANGED
@@ -394,7 +394,8 @@ def load_model(config, path, device="cpu"):
|
|
394 |
print(text_encoder)
|
395 |
config.cond_size = text_encoder.output_size
|
396 |
netG = NCSNpp(config)
|
397 |
-
print(netG)
|
|
|
398 |
ckpt = torch.load(path, map_location="cpu")
|
399 |
print("CK", ckpt)
|
400 |
for key in list(ckpt.keys()):
|
|
|
394 |
print(text_encoder)
|
395 |
config.cond_size = text_encoder.output_size
|
396 |
netG = NCSNpp(config)
|
397 |
+
#print(netG)
|
398 |
+
print(path, os.path.exists(path))
|
399 |
ckpt = torch.load(path, map_location="cpu")
|
400 |
print("CK", ckpt)
|
401 |
for key in list(ckpt.keys()):
|