search

wengong-jin / icml18-jtnn

Junction Tree Variational Autoencoder for Molecular Graph Generation (ICML 2018)
MIT License
509 stars 190 forks source link

is this a type error in jtnn_vae? #37

Closed asawanggaa closed 5 years ago

asawanggaa commented 5 years ago

line 91, self.stereo seems to be a bool type if self.stereo: stereo_loss, stereo_acc = self.stereo(mol_batch, mol_vec)

which self.stereo init here, line 46 self.stereo = stereo

wengong-jin commented 5 years ago

Thanks for pointing this out. The issue has been fixed.