helblazer811 commited on
Commit
f94da20
·
verified ·
1 Parent(s): 1adfc80

Release validated Zebra seed0 and simplify standard HF workflow

Browse files
README.md CHANGED
@@ -49,18 +49,19 @@ corresponding model folder from this repository.
49
  |---|---:|---|---:|---:|
50
  | sudoku | 0 | EqR Sudoku-Lite, 2,048 puzzles | 99.7070% (2042/2048) | 899.75 |
51
  | maze_unique | 0 | Official EqR Maze-Unique, 1,000 puzzles | 99.9% (999/1000) | 15.708 |
52
- | zebra | 1 | 5x5 Zebra, 1,024-puzzle seed0 subset | 99.4141% (1018/1024) | 4.0879 |
53
 
54
  Sudoku costs 0.8062 mean TFLOPs/puzzle; Maze-Unique costs 0.1781. Zebra FLOPs
55
- are unprofiled for this preprocessing, so only NFE is reported. Earlier Zebra SDE
56
- inference scored 99.8047% (1022/1024) on the same checkpoint; its 257-NFE held-t
57
- trajectory scores 99.7070%. The early-stop/restart policy trades some accuracy for
58
- substantially less compute. All Zebra test puzzles accept on the first attempt.
59
 
60
- These are selected-seed results, not seed averages. Maze-Unique restart seeds0/1/2:
61
  99.9%,98.5%,99.8%. Zebra SDE seeds0/1/2:99.6094%,99.8047%,99.5117%.
62
- Only one new Sudoku training seed was run. The Zebra checkpoint is selected using
63
- that recorded SDE evaluation, not a comparison of restart policies across seeds.
 
64
 
65
  ## Scope and reproducibility
66
 
 
49
  |---|---:|---|---:|---:|
50
  | sudoku | 0 | EqR Sudoku-Lite, 2,048 puzzles | 99.7070% (2042/2048) | 899.75 |
51
  | maze_unique | 0 | Official EqR Maze-Unique, 1,000 puzzles | 99.9% (999/1000) | 15.708 |
52
+ | zebra | 0 | 5x5 Zebra, 1,024-puzzle seed0 subset | 99.6094% (1020/1024) | 6.2402 |
53
 
54
  Sudoku costs 0.8062 mean TFLOPs/puzzle; Maze-Unique costs 0.1781. Zebra FLOPs
55
+ are unprofiled for this preprocessing, so only NFE is reported. Zebra seed0 scores99.8047% at257NFE with the fixed-depth held-t trajectory;
56
+ its stopping/restart policy scores99.6094% at6.2402meanNFE. Its earlier SDE
57
+ result is99.6094%. Every test puzzle accepts on the first attempt. FLOPs for this
58
+ 5x5 setup are unprofiled; NFE is reported instead.
59
 
60
+ All three hosted checkpoints now use training seed0. Maze-Unique restart seeds0/1/2:
61
  99.9%,98.5%,99.8%. Zebra SDE seeds0/1/2:99.6094%,99.8047%,99.5117%.
62
+ Only one new Sudoku training seed was run. Zebra seed0 was preferred for a
63
+ consistent release after its measured default score99.6094% exceeded seed1's
64
+ 99.4141%. Source checkpoint/step/seed details remain in each manifest.
65
 
66
  ## Scope and reproducibility
67
 
maze_unique/manifest.json CHANGED
@@ -19,5 +19,5 @@
19
  "calibration.npz": "3f0d8261ef2977c8a8d219dcb3f6cd146d3e7195438cfbe8939e774d1d8ff0dc",
20
  "source_evaluation.json": "cc68b4b64c9aa49b7a829ba36206dc05fd928668a349955b9263c3a1a4b4bcac"
21
  },
22
- "code_revision": "e18449e2c089e83f2069a750fe78aad091f2a3e3"
23
  }
 
19
  "calibration.npz": "3f0d8261ef2977c8a8d219dcb3f6cd146d3e7195438cfbe8939e774d1d8ff0dc",
20
  "source_evaluation.json": "cc68b4b64c9aa49b7a829ba36206dc05fd928668a349955b9263c3a1a4b4bcac"
21
  },
22
+ "code_revision": "e41dd277d4176fbc29b39219a1aef5c60a5481d8"
23
  }
release.json CHANGED
@@ -1,8 +1,13 @@
1
  {
2
  "code_repo": "helblazer811/FlowMapReasoning",
3
  "code_ref": "public-simplified",
4
- "code_revision": "e18449e2c089e83f2069a750fe78aad091f2a3e3",
5
  "package": "frm",
6
  "pypi_published": false,
7
- "scope": "private draft"
 
 
 
 
 
8
  }
 
1
  {
2
  "code_repo": "helblazer811/FlowMapReasoning",
3
  "code_ref": "public-simplified",
4
+ "code_revision": "e41dd277d4176fbc29b39219a1aef5c60a5481d8",
5
  "package": "frm",
6
  "pypi_published": false,
7
+ "scope": "private draft",
8
+ "training_seeds": {
9
+ "sudoku": 0,
10
+ "maze_unique": 0,
11
+ "zebra": 0
12
+ }
13
  }
sudoku/manifest.json CHANGED
@@ -19,5 +19,5 @@
19
  "calibration.npz": "e1f3b99cd5ce91dfe1e7de44995e1e38bbf1372766457025eca589625e89281e",
20
  "source_evaluation.json": "6085f2d71fe4aa39d253820326922de1236a142527fabf5746c8202269e0251c"
21
  },
22
- "code_revision": "e18449e2c089e83f2069a750fe78aad091f2a3e3"
23
  }
 
19
  "calibration.npz": "e1f3b99cd5ce91dfe1e7de44995e1e38bbf1372766457025eca589625e89281e",
20
  "source_evaluation.json": "6085f2d71fe4aa39d253820326922de1236a142527fabf5746c8202269e0251c"
21
  },
22
+ "code_revision": "e41dd277d4176fbc29b39219a1aef5c60a5481d8"
23
  }
verification.json CHANGED
@@ -1,24 +1,41 @@
1
  {
2
- "mode": "functional smoke, not full benchmark re-evaluation",
3
- "device": "cpu",
4
- "num_puzzles_per_task": 1,
5
- "installed_from": "GitHub/public-simplified",
6
- "package": "frm",
7
- "package_version": "0.2.0",
8
- "tested_code_revision": "5c40fd89631c5f7384589b408f6591dd98057621",
9
- "tested_bundle_revision": "3e8c45fd28b56f65c072334fc25c51328aa57850",
10
- "tasks": {
11
- "maze_unique": {
12
- "passed": true,
13
- "model_sha256": "3f0023f50352d395ca97a1d4a5c08a0f4bae4140fb53411a0a9c3840f7a7a4e3"
14
- },
15
- "sudoku": {
16
- "passed": true,
17
- "model_sha256": "5952ab8fb4184cffa7580b2035bc269531350d27e95cbdb4d26c9d5b241066c2"
18
- },
19
- "zebra": {
20
- "passed": true,
21
- "model_sha256": "2a94118070b94af73c9233ecf81400638ea5d3b4bc99b5b57169b796fcc847c4"
 
 
22
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
23
  }
24
  }
 
1
  {
2
+ "previous_download_smoke": {
3
+ "mode": "functional smoke, not full benchmark re-evaluation",
4
+ "device": "cpu",
5
+ "num_puzzles_per_task": 1,
6
+ "installed_from": "GitHub/public-simplified",
7
+ "package": "frm",
8
+ "package_version": "0.2.0",
9
+ "tested_code_revision": "5c40fd89631c5f7384589b408f6591dd98057621",
10
+ "tested_bundle_revision": "3e8c45fd28b56f65c072334fc25c51328aa57850",
11
+ "tasks": {
12
+ "maze_unique": {
13
+ "passed": true,
14
+ "model_sha256": "3f0023f50352d395ca97a1d4a5c08a0f4bae4140fb53411a0a9c3840f7a7a4e3"
15
+ },
16
+ "sudoku": {
17
+ "passed": true,
18
+ "model_sha256": "5952ab8fb4184cffa7580b2035bc269531350d27e95cbdb4d26c9d5b241066c2"
19
+ },
20
+ "zebra": {
21
+ "passed": true,
22
+ "model_sha256": "2a94118070b94af73c9233ecf81400638ea5d3b4bc99b5b57169b796fcc847c4"
23
+ }
24
  }
25
+ },
26
+ "standard_download_workflow": {
27
+ "code_revision": "700c1df787b00768bd3a3ef456a41a608ab2b343",
28
+ "method": "huggingface_hub.snapshot_download(local_dir) then python -m frm.evaluate",
29
+ "device": "cpu",
30
+ "task": "sudoku",
31
+ "num_puzzles": 1,
32
+ "passed": true
33
+ },
34
+ "zebra_seed0_gpu_validation": {
35
+ "job": 404162,
36
+ "num_puzzles": 1024,
37
+ "solved": 1020,
38
+ "mean_nfe": 6.240234375,
39
+ "model_sha256": "c43ce99d692dd8f0a15143bb28a35c69dd85f483ed430bca711ad6b865f7fe57"
40
  }
41
  }
zebra/expected_results.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
  "schema_version": 1,
3
  "git_commit": null,
4
- "checkpoint": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/hub_release_20261018_v2/bundles/zebra/model.pt",
5
- "checkpoint_sha256": "2a94118070b94af73c9233ecf81400638ea5d3b4bc99b5b57169b796fcc847c4",
6
- "benchmark": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/hub_release_20261018_v2/bundles/zebra/test.npz",
7
  "benchmark_sha256": "0948626c109bafdff92f16c8ea3a2c8a6078c39960c1261fed28d3a7f7204431",
8
  "policy_config": {
9
  "max_nfe": 257,
@@ -24,16 +24,16 @@
24
  "residual": "trailing-window mean of per-puzzle euclidean distance between consecutive denoiser distributions, averaged over non-clue cells",
25
  "oracle_used_for_test_selection": false,
26
  "kind": "residual_guided_inference_evaluation",
27
- "run_fingerprint": "950d7c7190c686a8b3c3269f58ec1a1e936f91db6448454553f81bd0485ee9ca",
28
  "num_puzzles": 1024,
29
  "max_samples_override": null,
30
- "model_step": 20000,
31
- "calibration": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/hub_release_20261018_v2/bundles/zebra/threshold.json",
32
- "calibration_sha256": "36c26f2a36371bd2d0ad0d3924a501c778cf6ee8112d8b3138770835440ec8b7",
33
  "calibration_benchmark_sha256": "b2c66c97d53152de289a2ab0c4ad2123ec5bed4bf53aaab14f21fa7f90c527cd",
34
- "threshold": 0.20256999680399107,
35
  "threshold_selection": {
36
- "threshold": 0.20256999680399107,
37
  "accuracy": 1.0,
38
  "solved": 256,
39
  "mean_nfe": 4.51171875,
@@ -45,26 +45,26 @@
45
  "policies": [
46
  {
47
  "policy": "Fixed-depth FRM",
48
- "solved": 1021,
49
  "num_puzzles": 1024,
50
- "solve_rate": 0.9970703125,
51
  "mean_nfe": 257.0,
52
  "mean_tflops": null
53
  },
54
  {
55
  "policy": "+ Residual early stopping",
56
- "solved": 1018,
57
  "num_puzzles": 1024,
58
- "solve_rate": 0.994140625,
59
- "mean_nfe": 4.087890625,
60
  "mean_tflops": null
61
  },
62
  {
63
  "policy": "+ Sequential restarts",
64
- "solved": 1018,
65
  "num_puzzles": 1024,
66
- "solve_rate": 0.994140625,
67
- "mean_nfe": 4.087890625,
68
  "mean_tflops": null
69
  }
70
  ],
@@ -76,22 +76,21 @@
76
  "attempted": 1024,
77
  "accepted": 1024,
78
  "remaining": 0,
79
- "mean_nfe_among_attempted": 4.087890625
80
  }
81
  ],
82
  "restart_frontier": [
83
  {
84
  "breadth_cap": 1,
85
- "solved": 1018,
86
  "num_puzzles": 1024,
87
- "solve_rate": 0.994140625,
88
- "mean_nfe": 4.087890625,
89
  "mean_tflops": null,
90
  "accepted": 1024,
91
  "unresolved": 0
92
  }
93
  ],
94
  "selection_rule": "stop each trajectory once the trailing-mean residual remains <= threshold for 1 checks and the decoded grid remains unchanged; restart only trajectories that reach max_nfe without acceptance; if max_breadth is exhausted, return the attempted terminal prediction with minimum residual",
95
- "labels_used_for": "final scoring only; never stopping, restart, or candidate selection",
96
- "compute_note": "FLOPs for this 5x5 model have not been profiled; use measured NFE."
97
  }
 
1
  {
2
  "schema_version": 1,
3
  "git_commit": null,
4
+ "checkpoint": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/zebra_seed0_release_20261018_v1/bundles/zebra/model.pt",
5
+ "checkpoint_sha256": "c43ce99d692dd8f0a15143bb28a35c69dd85f483ed430bca711ad6b865f7fe57",
6
+ "benchmark": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/zebra_seed0_release_20261018_v1/bundles/zebra/test.npz",
7
  "benchmark_sha256": "0948626c109bafdff92f16c8ea3a2c8a6078c39960c1261fed28d3a7f7204431",
8
  "policy_config": {
9
  "max_nfe": 257,
 
24
  "residual": "trailing-window mean of per-puzzle euclidean distance between consecutive denoiser distributions, averaged over non-clue cells",
25
  "oracle_used_for_test_selection": false,
26
  "kind": "residual_guided_inference_evaluation",
27
+ "run_fingerprint": "ad3a2420cab1679dab9f04d8f963fc6acf0bfce272b1c233e0e0d84b10810112",
28
  "num_puzzles": 1024,
29
  "max_samples_override": null,
30
+ "model_step": 115000,
31
+ "calibration": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/zebra_seed0_release_20261018_v1/bundles/zebra/threshold.json",
32
+ "calibration_sha256": "d7401816ce4e2638c587f5793f55b60a1110acb03a87941f71b1b05aa7bb46c3",
33
  "calibration_benchmark_sha256": "b2c66c97d53152de289a2ab0c4ad2123ec5bed4bf53aaab14f21fa7f90c527cd",
34
+ "threshold": 0.10050372427701933,
35
  "threshold_selection": {
36
+ "threshold": 0.10050372427701933,
37
  "accuracy": 1.0,
38
  "solved": 256,
39
  "mean_nfe": 4.51171875,
 
45
  "policies": [
46
  {
47
  "policy": "Fixed-depth FRM",
48
+ "solved": 1022,
49
  "num_puzzles": 1024,
50
+ "solve_rate": 0.998046875,
51
  "mean_nfe": 257.0,
52
  "mean_tflops": null
53
  },
54
  {
55
  "policy": "+ Residual early stopping",
56
+ "solved": 1020,
57
  "num_puzzles": 1024,
58
+ "solve_rate": 0.99609375,
59
+ "mean_nfe": 6.240234375,
60
  "mean_tflops": null
61
  },
62
  {
63
  "policy": "+ Sequential restarts",
64
+ "solved": 1020,
65
  "num_puzzles": 1024,
66
+ "solve_rate": 0.99609375,
67
+ "mean_nfe": 6.240234375,
68
  "mean_tflops": null
69
  }
70
  ],
 
76
  "attempted": 1024,
77
  "accepted": 1024,
78
  "remaining": 0,
79
+ "mean_nfe_among_attempted": 6.240234375
80
  }
81
  ],
82
  "restart_frontier": [
83
  {
84
  "breadth_cap": 1,
85
+ "solved": 1020,
86
  "num_puzzles": 1024,
87
+ "solve_rate": 0.99609375,
88
+ "mean_nfe": 6.240234375,
89
  "mean_tflops": null,
90
  "accepted": 1024,
91
  "unresolved": 0
92
  }
93
  ],
94
  "selection_rule": "stop each trajectory once the trailing-mean residual remains <= threshold for 1 checks and the decoded grid remains unchanged; restart only trajectories that reach max_nfe without acceptance; if max_breadth is exhausted, return the attempted terminal prediction with minimum residual",
95
+ "labels_used_for": "final scoring only; never stopping, restart, or candidate selection"
 
96
  }
zebra/manifest.json CHANGED
@@ -1,25 +1,25 @@
1
  {
2
  "task": "zebra",
3
- "training_seed": 1,
4
- "selection": "best measured seed among recorded runs; see results table",
5
- "source_checkpoint": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/runs/20261015-1119/full/zebra_s1/stage_b/best.pt",
6
- "source_checkpoint_sha256": "e1e8c364539adf482d04d0de4e234e7a1bfc71e55547824d2dd5789f85fa0ef4",
7
- "source_step": 20000,
8
- "checkpoint_sha256": "2a94118070b94af73c9233ecf81400638ea5d3b4bc99b5b57169b796fcc847c4",
9
  "checkpoint_format": "raw model plus EMA/config/provenance; optimizer omitted",
10
  "scope": "5x5 Zebra, repacked answers (not paper mixed-size pinned)",
 
11
  "files": {
12
- "training.yaml": "29d7da5265a805190959d234de1f3702591c15632b80ac713f3744706147d3c2",
13
- "model.pt": "2a94118070b94af73c9233ecf81400638ea5d3b4bc99b5b57169b796fcc847c4",
14
- "provenance.json": "476fdce0c4bdd67e05c3e7554c9a6f99a9b54406eac859cc59ca48dc293e8974",
15
- "expected_results.json": "46a370d09021d2f900475851296b17fcd7fbe332dd5b4ac48fe7878dcf051cfd",
16
- "test.npz": "0948626c109bafdff92f16c8ea3a2c8a6078c39960c1261fed28d3a7f7204431",
17
- "threshold.json": "36c26f2a36371bd2d0ad0d3924a501c778cf6ee8112d8b3138770835440ec8b7",
18
- "vocab.json": "17c8b60b8a2ec9feba73a0d6f5cd5f973da60c54d0d82b23a61743c27715698d",
19
  "calibration.npz": "b2c66c97d53152de289a2ab0c4ad2123ec5bed4bf53aaab14f21fa7f90c527cd",
 
20
  "meta.json": "1d19448cf8bc66d7bb87d8d5ac73367cc10c61eac77acf5172ba1c5dcd10130c",
21
- "source_evaluation.json": "8466835dc4e90919c24cfefcc5de98814544e14c42b9f076631fda0d97397375"
 
 
 
 
 
 
22
  },
23
- "calibration_note": "256 unique 5x5 training rows, seed20260910, explicitly excludes test puzzles; not the mixed-size paper policy.",
24
- "code_revision": "e18449e2c089e83f2069a750fe78aad091f2a3e3"
25
  }
 
1
  {
2
  "task": "zebra",
3
+ "training_seed": 0,
4
+ "selection": "seed0 release for consistency; see complete seed results",
5
+ "source_checkpoint": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/runs/20261015-1119/full/zebra_s0/stage_b/best.pt",
6
+ "source_checkpoint_sha256": "c2c566d99eff64788bfbbfb8dc4571a6f0b451f136fffd772298e56ac11c80cd",
7
+ "source_step": 115000,
8
+ "checkpoint_sha256": "c43ce99d692dd8f0a15143bb28a35c69dd85f483ed430bca711ad6b865f7fe57",
9
  "checkpoint_format": "raw model plus EMA/config/provenance; optimizer omitted",
10
  "scope": "5x5 Zebra, repacked answers (not paper mixed-size pinned)",
11
+ "calibration_note": "256 unique 5x5 training rows, seed20260910, excludes test puzzles",
12
  "files": {
 
 
 
 
 
 
 
13
  "calibration.npz": "b2c66c97d53152de289a2ab0c4ad2123ec5bed4bf53aaab14f21fa7f90c527cd",
14
+ "vocab.json": "17c8b60b8a2ec9feba73a0d6f5cd5f973da60c54d0d82b23a61743c27715698d",
15
  "meta.json": "1d19448cf8bc66d7bb87d8d5ac73367cc10c61eac77acf5172ba1c5dcd10130c",
16
+ "model.pt": "c43ce99d692dd8f0a15143bb28a35c69dd85f483ed430bca711ad6b865f7fe57",
17
+ "expected_results.json": "80a00058c2e5edb8ed4bacb9d469bcdf41050554266c80d36e4600e575d8ece9",
18
+ "provenance.json": "0895404bd22920def2dcccdf0d3fc0af3e11efaa4d65c285a3f68e3def6a6b9b",
19
+ "test.npz": "0948626c109bafdff92f16c8ea3a2c8a6078c39960c1261fed28d3a7f7204431",
20
+ "training.yaml": "83826d69f8f110cb4b5b6f4477c2a1578524c2326d05b4fd160f8c733dd6f2db",
21
+ "threshold.json": "d7401816ce4e2638c587f5793f55b60a1110acb03a87941f71b1b05aa7bb46c3",
22
+ "source_evaluation.json": "68e5262551fd44859df69376119baf15e68ed5eb368a34e96d61f6fcbb17ffd9"
23
  },
24
+ "code_revision": "e41dd277d4176fbc29b39219a1aef5c60a5481d8"
 
25
  }
zebra/model.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2a94118070b94af73c9233ecf81400638ea5d3b4bc99b5b57169b796fcc847c4
3
- size 200110615
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c43ce99d692dd8f0a15143bb28a35c69dd85f483ed430bca711ad6b865f7fe57
3
+ size 200110743
zebra/provenance.json CHANGED
@@ -1,11 +1,11 @@
1
  {
2
  "task": "zebra",
3
- "training_seed": 1,
4
- "selection": "best measured seed among recorded runs; see results table",
5
- "source_checkpoint": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/runs/20261015-1119/full/zebra_s1/stage_b/best.pt",
6
- "source_checkpoint_sha256": "e1e8c364539adf482d04d0de4e234e7a1bfc71e55547824d2dd5789f85fa0ef4",
7
- "source_step": 20000,
8
- "checkpoint_sha256": "2a94118070b94af73c9233ecf81400638ea5d3b4bc99b5b57169b796fcc847c4",
9
  "checkpoint_format": "raw model plus EMA/config/provenance; optimizer omitted",
10
  "scope": "5x5 Zebra, repacked answers (not paper mixed-size pinned)"
11
  }
 
1
  {
2
  "task": "zebra",
3
+ "training_seed": 0,
4
+ "selection": "seed0 release for consistency; see complete seed results",
5
+ "source_checkpoint": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/runs/20261015-1119/full/zebra_s0/stage_b/best.pt",
6
+ "source_checkpoint_sha256": "c2c566d99eff64788bfbbfb8dc4571a6f0b451f136fffd772298e56ac11c80cd",
7
+ "source_step": 115000,
8
+ "checkpoint_sha256": "c43ce99d692dd8f0a15143bb28a35c69dd85f483ed430bca711ad6b865f7fe57",
9
  "checkpoint_format": "raw model plus EMA/config/provenance; optimizer omitted",
10
  "scope": "5x5 Zebra, repacked answers (not paper mixed-size pinned)"
11
  }
zebra/source_evaluation.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "dataset": "zebra",
3
- "checkpoint": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/runs/20261015-1119/full/zebra_s1/stage_b/best.pt",
4
- "checkpoint_step": 20000,
5
  "use_ema": true,
6
  "protocol": {
7
  "split": "test",
@@ -19,15 +19,15 @@
19
  },
20
  "metrics": {
21
  "n_samples": 1024,
22
- "exact_match": 0.998046875,
23
- "solved": 0.998046875,
24
- "token_accuracy": 0.999140625,
25
  "clue_consistency": 1.0
26
  },
27
  "checkpoint_provenance": {
28
  "dataset": "zebra",
29
  "stage": "b",
30
- "seed": 1,
31
  "recipe": {
32
  "data": {
33
  "dataset": "zebra",
@@ -120,7 +120,7 @@
120
  "self_cond": true
121
  },
122
  "pipeline": {
123
- "seed": 1,
124
  "num_workers": 2,
125
  "stage_a_steps": 200000,
126
  "stage_b_steps": 200000,
 
1
  {
2
  "dataset": "zebra",
3
+ "checkpoint": "/proj/grp-flow-map/LoopedFlows/experiments/public_simplified_repro/outputs/runs/20261015-1119/full/zebra_s0/stage_b/best.pt",
4
+ "checkpoint_step": 115000,
5
  "use_ema": true,
6
  "protocol": {
7
  "split": "test",
 
19
  },
20
  "metrics": {
21
  "n_samples": 1024,
22
+ "exact_match": 0.99609375,
23
+ "solved": 0.99609375,
24
+ "token_accuracy": 0.9985546875,
25
  "clue_consistency": 1.0
26
  },
27
  "checkpoint_provenance": {
28
  "dataset": "zebra",
29
  "stage": "b",
30
+ "seed": 0,
31
  "recipe": {
32
  "data": {
33
  "dataset": "zebra",
 
120
  "self_cond": true
121
  },
122
  "pipeline": {
123
+ "seed": 0,
124
  "num_workers": 2,
125
  "stage_a_steps": 200000,
126
  "stage_b_steps": 200000,
zebra/threshold.json CHANGED
@@ -2,7 +2,7 @@
2
  "schema_version": 1,
3
  "git_commit": null,
4
  "checkpoint": "model.pt",
5
- "checkpoint_sha256": "2a94118070b94af73c9233ecf81400638ea5d3b4bc99b5b57169b796fcc847c4",
6
  "benchmark": "calibration.npz",
7
  "benchmark_sha256": "b2c66c97d53152de289a2ab0c4ad2123ec5bed4bf53aaab14f21fa7f90c527cd",
8
  "policy_config": {
@@ -26,10 +26,10 @@
26
  "kind": "residual_stopping_calibration",
27
  "num_puzzles": 256,
28
  "max_samples_override": null,
29
- "model_step": 20000,
30
  "labels_used_for": "development-set threshold calibration only",
31
  "selected": {
32
- "threshold": 0.20256999680399107,
33
  "accuracy": 1.0,
34
  "solved": 256,
35
  "mean_nfe": 4.51171875,
 
2
  "schema_version": 1,
3
  "git_commit": null,
4
  "checkpoint": "model.pt",
5
+ "checkpoint_sha256": "c43ce99d692dd8f0a15143bb28a35c69dd85f483ed430bca711ad6b865f7fe57",
6
  "benchmark": "calibration.npz",
7
  "benchmark_sha256": "b2c66c97d53152de289a2ab0c4ad2123ec5bed4bf53aaab14f21fa7f90c527cd",
8
  "policy_config": {
 
26
  "kind": "residual_stopping_calibration",
27
  "num_puzzles": 256,
28
  "max_samples_override": null,
29
+ "model_step": 115000,
30
  "labels_used_for": "development-set threshold calibration only",
31
  "selected": {
32
+ "threshold": 0.10050372427701933,
33
  "accuracy": 1.0,
34
  "solved": 256,
35
  "mean_nfe": 4.51171875,
zebra/training.yaml CHANGED
@@ -22,7 +22,7 @@ train:
22
  val_every: 5000
23
  save_best_metric: val/solved
24
  pipeline:
25
- seed: 1
26
  num_workers: 2
27
  stage_a_steps: 200000
28
  stage_b_steps: 200000
 
22
  val_every: 5000
23
  save_best_metric: val/solved
24
  pipeline:
25
+ seed: 0
26
  num_workers: 2
27
  stage_a_steps: 200000
28
  stage_b_steps: 200000