Make sure report works without GPU

mila-iqia · Feb 26, 2024 · a7f2d86 · a7f2d86
1 parent 629bee3
commit a7f2d86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/milabench/config.py b/milabench/config.py
@@ -175,7 +175,7 @@ def resolve_addresses(nodes):
 
 
 def get_gpu_capacity():
-    capacity = float("+inf")
+    capacity = float(0)
 
     for k, v in get_gpu_info()["gpus"].items():
         capacity = min(v["memory"]["total"], capacity)