Changed the GPU call to CPU

kumaranu · Mar 12, 2024 · 7cb7ac9 · 7cb7ac9
1 parent fde162a
commit 7cb7ac9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/calcs.py b/src/calcs.py
@@ -4,8 +4,10 @@
 def calc():
 
     mlcalculator = MACECalculator(
-        model_paths='/global/home/users/kumaranu/Documents/gpu_jobs/MACE_model.model',
-        device='cuda',
+        #model_paths='/global/home/users/kumaranu/Documents/gpu_jobs/MACE_model.model',
+        model_paths='/global/home/users/kumaranu/Documents/gpu_jobs/MACE_model_cpu.model',
+        #device='cuda',
+        device='cpu',
         default_dtype="float64",
     )
     return mlcalculator