Mehdi Cherti commited on
Commit
c4218ab
1 Parent(s): b195df1
Files changed (1) hide show
  1. test_ddgan.py +1 -0
test_ddgan.py CHANGED
@@ -389,6 +389,7 @@ class ObjectFromDict:
389
 
390
  def load_model(config, path, device="cpu"):
391
  config = ObjectFromDict(config)
 
392
  text_encoder = build_encoder(name=config.text_encoder, masked_mean=config.masked_mean)
393
  print(text_encoder)
394
  config.cond_size = text_encoder.output_size
 
389
 
390
  def load_model(config, path, device="cpu"):
391
  config = ObjectFromDict(config)
392
+ print(config)
393
  text_encoder = build_encoder(name=config.text_encoder, masked_mean=config.masked_mean)
394
  print(text_encoder)
395
  config.cond_size = text_encoder.output_size