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 error:( #4

Closed xnuohz closed 3 years ago

xnuohz commented 3 years ago

Hi kexin, I used your code to do regression task on KIBA dataset, and it got OSError: [Errno 12] Cannot allocate memory, when I change num_workers to 0, it may be ok, but too slow. Is there any other ways to solve this problem? Thanks:)

kexinhuang12345 commented 3 years ago

Hi, the kiba dataset is not supported in MolTrans. Are you running your own dataset? It should be okay, since bindingdb is of similar order of size as kiba, and it works fine.