From a7f2d860e19fe155f1507607a2c851d92a99e632 Mon Sep 17 00:00:00 2001 From: Pierre Delaunay Date: Mon, 26 Feb 2024 11:45:22 -0500 Subject: [PATCH] Make sure report works without GPU --- milabench/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/milabench/config.py b/milabench/config.py index fa3b85f47..14c8bf46e 100644 --- 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)