From 702c1c6b97fd52b903920c934f4631e7161e1147 Mon Sep 17 00:00:00 2001 From: Setepenre Date: Fri, 7 Jun 2024 17:12:31 -0400 Subject: [PATCH] Update sizer.py --- milabench/sizer.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/milabench/sizer.py b/milabench/sizer.py index 31e6c6d20..2f7b0654d 100644 --- a/milabench/sizer.py +++ b/milabench/sizer.py @@ -106,7 +106,12 @@ def auto_size(self, benchmark, capacity): return None config = self.benchscaling(benchmark) - + model = config.get("model", None) + + if model is None: + print(f"Missing batch-size model for {benchmark}") + return 1 + data = list(sorted(config["model"].items(), key=lambda x: x[0])) mem = [to_octet(v[1]) for v in data] size = [float(v[0]) for v in data]