Update eval.py

ChEB-AI · Dec 6, 2024 · 6e9e098 · 6e9e098
1 parent 6ba3921
commit 6e9e098
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eval.py b/eval.py
@@ -12,7 +12,7 @@
 import torch
 import pickle
 
-DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+DEVICE = torch.device("cuda:1" if torch.cuda.is_available() else "cpu")
 print(DEVICE)