refactoring training spc setup and benchmarking

2026-05-31 16:43:36 +00:00 · 2026-03-08 18:30:53 +01:00
parent 9fafb26ec8
commit 73246d7dd8
36 changed files with 2180 additions and 613 deletions
--- a/scripts/tpu_vm_sweep_agent.py
+++ b/scripts/tpu_vm_sweep_agent.py
@@ -96,7 +96,11 @@ def _extract_metrics(output: str) -> dict:
            obj = json.loads(block)
        except Exception:
            continue
-        if isinstance(obj, dict) and ("sweep/score" in obj or "eval/reward" in obj):
+        if isinstance(obj, dict) and (
+            "objective/score" in obj
+            or "eval/reward_mean" in obj
+            or "sweep/score" in obj
+        ):
            return obj
    return {}