runtime training the zinc dataset

aspuru-guzik-group / chemical_vae

Code for 10.1021/acscentsci.7b00572, now running on Keras 2.0 and Tensorflow

Apache License 2.0

483 stars 180 forks source link

runtime training the zinc dataset #24

Closed jbdav closed 4 years ago

jbdav commented 4 years ago

Hello,

what is the average runtime for training the zince dataset (70 epochs)? I'm running this on a dual GPU workstation and a single epoch takes a very long time. Are there additional parameter settings I need to change?

Thanks!

jbdav commented 4 years ago

tensorflow-gpu upgrade