From 7d36feec44082cf5db841d79fc5b02ba5d5f8c32 Mon Sep 17 00:00:00 2001 From: Yann Ahlgrim Date: Sat, 20 Jun 2026 11:56:19 +0200 Subject: [PATCH] handle imagenet models --- tools/summarize_grid.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/tools/summarize_grid.py b/tools/summarize_grid.py index aaa2c43..09845c7 100644 --- a/tools/summarize_grid.py +++ b/tools/summarize_grid.py @@ -200,6 +200,15 @@ def _infer_model_from_run_name(run_name): return None +def _normalize_model_type(model_name): + parts = model_name.split("-") + base = parts[0] + for part in parts[1:]: + if part.startswith("in"): + return f"{base}-{part}" + return base + + def _get_model_name(dirpath, run_name=None): params = _load_params_simple(dirpath) name = None @@ -207,7 +216,9 @@ def _get_model_name(dirpath, run_name=None): name = _get_in_params(params, "meta.model_name") if name is None and run_name: name = _infer_model_from_run_name(run_name) - return name or "unknown" + if name: + return _normalize_model_type(name) + return "unknown" def _collect_rows(root_dir, metric_key, col_paths):