DiffSBDD
DiffSBDD copied to clipboard
High CUDA memory usage while generating ligands
Hi all,
When I ran generate_ligands.py with parameters "--sanitize --relax --num_nodes_lig 50 --n_samples 20" in moad_fullatom_cond model. It took 5368MB CUDA memory!!!
Is it OK for only sampling 20 molecules with high CUDA memory usage? And, how can I generate 10 thousand or more molecules?