Mehdi Cherti commited on
Commit
9916e2b
1 Parent(s): d8a4058
Files changed (1) hide show
  1. 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()):