Update lp.py

martinjingyu · Dec 10, 2023 · 0907a23 · 0907a23
1 parent bd78a04
commit 0907a23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lp.py b/lp.py
@@ -34,7 +34,7 @@ def load(emb_file_path):
                 emb_dict[index] = np.array(emb.split()).astype(np.float32)
 
     return train_para, emb_dict  
-device = torch.device("cuda:1")
+device = torch.device("cuda:0")
 emb_file_path = './PubMed/emb.dat'
 train_para, emb_dict = load(emb_file_path)
 link_test_file = './PubMed/sample.dat'