update

test_ddgan.py

@@ -390,9 +390,12 @@ class ObjectFromDict:
 def load_model(config, path, device="cpu"):
     config = ObjectFromDict(config)
     text_encoder = build_encoder(name=config.text_encoder, masked_mean=config.masked_mean)
+    print(text_encoder)
     config.cond_size = text_encoder.output_size
     netG = NCSNpp(config)
+    print(netG)
     ckpt = torch.load(path, map_location="cpu")
+    print("CK", ckpt)
     for key in list(ckpt.keys()):
         if key.startswith("module"):
             ckpt[key[7:]] = ckpt.pop(key)