search

kexinhuang12345 / MolTrans

MolTrans: Molecular Interaction Transformer for Drug Target Interaction Prediction (Bioinformatics)
https://academic.oup.com/bioinformatics/advance-article/doi/10.1093/bioinformatics/btaa880/5929692
BSD 3-Clause "New" or "Revised" License
175 stars 43 forks source link

Running MolTrans without GPU #10

Closed waqarkaleemkhan closed 3 years ago

waqarkaleemkhan commented 3 years ago

hi @kexinhuang12345 hope you will be doing well I want to run your code can you please guide how to run it without using GPU because I don't have GPU on my laptop and for information I want to run the train.py file, not the experiment notebook. Thanks

kexinhuang12345 commented 3 years ago

Hey, sorry, this model uses transformers and large CNNs, so running on CPU would take forever and are thus highly not recommended. For quick DTI prediction on CPUs, please checkout our DeepPurpose: https://github.com/kexinhuang12345/DeepPurpose