Error in loading state_dict for DiffVAE

wengong-jin / iclr19-graph2graph

Learning Multimodal Graph-to-Graph Translation for Molecular Optimization (ICLR 2019)

MIT License

147 stars 40 forks source link

Error in loading state_dict for DiffVAE #5

Closed SejeongPark8354 closed 4 years ago

SejeongPark8354 commented 4 years ago

First of all, thank you for your wonderful study in molecular generation. I am learning a lot from your research. When I try testing process like below figure, error in loading state_dic was occurred. (same problem: model.iter-5 in logp04, logp06 in diff_vae, model.iter-3 in qed in diff_vae_gan, etc.) Is there any solution to this problem? I'll wait your help. Thanks.

wengong-jin commented 4 years ago

Sorry for my late reply. Please run the script with --hidden_size 330