search

wengong-jin / icml18-jtnn

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

update line data[0] #45

Open abhik1368 opened 5 years ago

abhik1368 commented 5 years ago

Line 166 and 261 was throwing an error invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number .

update line data[0] to data