Use standard Hugging Face downloads and local evaluation
Browse files
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 |
-
##
|
| 16 |
|
| 17 |
-
GitHub installation only; nothing
|
| 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 #
|
| 24 |
-
frm-eval --task sudoku
|
| 25 |
-
frm-eval --task maze_unique
|
| 26 |
-
frm-eval --task zebra
|
| 27 |
```
|
| 28 |
|
| 29 |
-
|
|
|
|
|
|
|
| 30 |
|
| 31 |
```python
|
| 32 |
-
from
|
| 33 |
-
|
| 34 |
-
|
|
|
|
|
|
|
|
|
|
| 35 |
```
|
| 36 |
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
`--max-samples 8` for a functional
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 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 |
-
`
|
| 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).
|