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

Use standard Hugging Face downloads and local evaluation

Browse files
Files changed (1) hide show
  1. README.md +19 -18
README.md CHANGED
@@ -12,35 +12,36 @@ tags:
12
  One selected checkpoint per task, with portable benchmark/calibration assets and
13
  measured results. This repository stays private during testing.
14
 
15
- ## Install and evaluate
16
 
17
- GitHub installation only; nothing has been published to PyPI. A CUDA GPU is recommended.
18
 
19
  ```bash
20
  python -m venv .venv
21
  source .venv/bin/activate
22
  pip install "git+https://github.com/helblazer811/FlowMapReasoning.git@public-simplified"
23
- hf auth login # private repository: use an authorized account
24
- frm-eval --task sudoku
25
- frm-eval --task maze_unique
26
- frm-eval --task zebra
27
  ```
28
 
29
- Or from Python:
 
 
30
 
31
  ```python
32
- from frm import evaluate
33
- result = evaluate("sudoku", output_dir="outputs/sudoku")
34
- print(result["policies"][-1])
 
 
 
35
  ```
36
 
37
- All model files, benchmark arrays and calibration are downloaded automatically and
38
- verified by SHA-256. The default is calibrated sequential restarts. Add
39
- `--max-samples 8` for a functional smoke test, not a reported benchmark score.
40
- Use `--revision <Hub commit>` to pin the bundle. Outputs resume automatically.
41
- The evaluator computes a fixed-depth comparison too: evaluation wall-clock work
42
- includes that extra baseline; reported NFE/FLOPs describe the selected restart
43
- policy, not the total cost of generating all three comparison rows.
44
 
45
  ## Selected models and measured results
46
 
@@ -77,7 +78,7 @@ Each folder contains `model.pt`, `training.yaml`, `threshold.json`, `test.npz`,
77
  The manifest records the original checkpoint hash and the compact inference-checkpoint
78
  hash. Optimizer state is omitted; raw weights and EMA are retained. These are compact PyTorch checkpoints from the recorded training runs.
79
  For training from scratch, download `training.yaml` and run
80
- `frm-train --config training.yaml --seed <recorded seed>` on the matching data.
81
 
82
  Code is being drafted on
83
  [public-simplified](https://github.com/helblazer811/FlowMapReasoning/tree/public-simplified).
 
12
  One selected checkpoint per task, with portable benchmark/calibration assets and
13
  measured results. This repository stays private during testing.
14
 
15
+ ## Setup
16
 
17
+ GitHub installation only; nothing is published to PyPI. A CUDA GPU is recommended.
18
 
19
  ```bash
20
  python -m venv .venv
21
  source .venv/bin/activate
22
  pip install "git+https://github.com/helblazer811/FlowMapReasoning.git@public-simplified"
23
+ hf auth login # while this model repository is private
 
 
 
24
  ```
25
 
26
+ ## Download and evaluate
27
+
28
+ Use standard Hugging Face downloads:
29
 
30
  ```python
31
+ from huggingface_hub import snapshot_download
32
+ snapshot_download("helblazer811/FlowReasoningModel", allow_patterns=["sudoku/*"], local_dir="models")
33
+ ```
34
+
35
+ ```bash
36
+ python -m frm.evaluate --model-dir models/sudoku
37
  ```
38
 
39
+ Replace `sudoku` with `maze_unique` or `zebra` for the other models. Evaluation
40
+ checks the downloaded files and uses calibrated sequential restarts. Results go
41
+ under `outputs/`. Add `--max-samples 8` for a functional check, not a benchmark score.
42
+ The evaluator computes fixed-depth comparisons too; reported policy NFE excludes
43
+ that additional comparison work. For manual download, obtain all files in the
44
+ corresponding model folder from this repository.
 
45
 
46
  ## Selected models and measured results
47
 
 
78
  The manifest records the original checkpoint hash and the compact inference-checkpoint
79
  hash. Optimizer state is omitted; raw weights and EMA are retained. These are compact PyTorch checkpoints from the recorded training runs.
80
  For training from scratch, download `training.yaml` and run
81
+ `python -m frm.train --config training.yaml` on the matching data.
82
 
83
  Code is being drafted on
84
  [public-simplified](https://github.com/helblazer811/FlowMapReasoning/tree/public-simplified).