search

zhanglabNKU / DAttProt

5 stars 0 forks source link

training data on CPU? #1

Open BinhongLiu opened 2 years ago

BinhongLiu commented 2 years ago

Hi, I'm trying to pre-training the dataset, but it seems that GPU is needed in the pretrain.py script. I'm not sure how to modify the script so that it can just use CPU to conduct the analysis. Could you help me with this and provide a modified CPU version of the DAttProt scripts? Thanks

Nanohemu commented 1 year ago

Maybe you can try this: In line 34: device = torch.device("cpu")