budget summary handles aligned spec

ptr/flops.py

@@ -35,12 +35,12 @@ def step_flops(model, S, batch, n_iters, training=True):
 
 def flops_budget_summary(model, S, batch, n_iters, steps):
     """Total training FLOPs for one run, plus params. `n_iters` may be an int
-    or ('dynamic', lo, hi) for per-batch sampled unrolls (mean used)."""
+    or a (mode, lo, hi) tuple for per-batch sampled unrolls (mean used)."""
     p = count_params(model)
     if isinstance(n_iters, tuple):
         lo, hi = n_iters[1], n_iters[2]
         mean_iters = (lo + hi) / 2.0
-        desc = f"dynamic U{{{lo}..{hi}}}"
+        desc = f"{n_iters[0]} U{{{lo}..{hi}}}"
     else:
         mean_iters = n_iters
         desc = str(n_iters)