stage4: aligned at 100k steps

run_all.py

@@ -132,6 +132,18 @@ def stage3(args):
         run_eval(args, f"out/{name}.pt")
 
 
+def stage4(args):
+    """Aligned objective at 4x the step budget (1 seed): distinguishes
+    'harder optimization, needs more steps' from 'fundamentally harder'
+    after the 25k-step aligned runs plateaued near 0.10 id_acc."""
+    lr = chosen_lr(args.hub_repo)
+    name = "rec1aligned-long-seed0"
+    if not os.path.exists(f"out/{name}.pt"):
+        run(train_cmd({"arch": "rec", "depth": 1, "iters": "aligned",
+                       "seed": 0}, name, 4 * STEPS, lr, args.hub_repo))
+    run_eval(args, f"out/{name}.pt")
+
+
 def stage_eval(args):
     if not glob.glob("out/*.pt"):
         # recovery mode (fresh container): pull every checkpoint from the Hub,
@@ -153,13 +165,14 @@ def stage_eval(args):
 def main():
     p = argparse.ArgumentParser()
     p.add_argument("stage", choices=["lrsweep", "stage1", "stage2", "stage3",
-                                     "eval"])
+                                     "stage4", "eval"])
     p.add_argument("--hub-repo", default=None)
     a = p.parse_args()
     if a.hub_repo:
         os.environ.setdefault("TRACKIO_HUB_REPO", a.hub_repo)
     {"lrsweep": stage_lrsweep, "stage1": stage1, "stage2": stage2,
-     "stage3": stage3, "eval": stage_eval}[a.stage](a)
+     "stage3": stage3, "stage4": stage4,
+     "eval": stage_eval}[a.stage](a)
 
 
 if __name__ == "__main__":