search

wengong-jin / icml18-jtnn

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

change data[0] to data.item() #42

Open asawanggaa opened 5 years ago

asawanggaa commented 5 years ago

every data[0] should be change to data.item() while data tensor has only one element, this make /jtnn/jtnn_vae.py or other files can work when stereo is valid.