solved bug for assigning trainer device

CederGroupHub · Feb 22, 2024 · c8c1351 · c8c1351
1 parent af44916
commit c8c1351
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chgnet/trainer/trainer.py b/chgnet/trainer/trainer.py
@@ -188,7 +188,7 @@ def __init__(
             self.device = "cpu"
         if self.device == "cuda":
             # Determine cuda device with most available memory
-            device_with_most_available_memory = cuda_devices_sorted_by_free_mem()[0]
+            device_with_most_available_memory = cuda_devices_sorted_by_free_mem()[-1]
             self.device = f"cuda:{device_with_most_available_memory}"
 
         self.print_freq = print_freq