search

topazape / LSTM_Chem

Implementation of the paper - Generative Recurrent Networks for De Novo Drug Design.
The Unlicense
116 stars 55 forks source link

About GPU memory usage? #24

Open xpyqiubai opened 2 years ago

xpyqiubai commented 2 years ago

Thank the author for the code.

I'd like to ask you some questions: 1.The memory capacity and GPU model of your computer. 2.GPU memory footprint while training code.

Looking forward to your reply. I want to know if my GTX1060 GPU can train with this code.

Heartfelt thanks.

topazape commented 2 years ago

When I implemented this code, I used GTX TITAN X / 12GB GDDR5. If you are warned that your GPU is low on memory, please set a smaller batch size in base_config.json.