diff options
| author | YurenHao0426 <blackhao0426@gmail.com> | 2026-06-13 12:35:36 -0500 |
|---|---|---|
| committer | YurenHao0426 <blackhao0426@gmail.com> | 2026-06-13 12:35:36 -0500 |
| commit | 66e0d8b9fd4d0f7a2231d689c055e26fdf1cf04a (patch) | |
| tree | c29cba61124018755a19b02c9d33e3ad5f2e05cc /PROVENANCE.md | |
Curated export for clone-and-run Maze training (2x A6000) + diagnostics.
trm/hrm pretrain.py carry trajectory-augmentation code (backward-compatible).
Heavy artifacts (checkpoints/wandb/npz) gitignored; see PROVENANCE.md.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Diffstat (limited to 'PROVENANCE.md')
| -rw-r--r-- | PROVENANCE.md | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/PROVENANCE.md b/PROVENANCE.md new file mode 100644 index 0000000..33ddc9b --- /dev/null +++ b/PROVENANCE.md @@ -0,0 +1,16 @@ +# Provenance + +This repo is a curated export of the `rrm` research workspace. `trm/`, `hrm/`, `srm/` are +upstream clones with our modifications committed as plain files (nested `.git` not included). + +| dir | upstream | pinned commit | our changes | +|---|---|---|---| +| `trm/` | github.com/SamsungSAILMontreal/TinyRecursiveModels | `c01103738605ba39d1430519b1ee0c62f4c707f8` | `pretrain.py` +625 lines: trajectory-augmentation training (multi-rollout perturbation, directional FD variants). Backward-compatible — `trajectory_augment` defaults False, so the official Maze/Sudoku recipes run unchanged. | +| `hrm/` | github.com/sapientinc/HRM | `ac15626f8db096a63c775b84c9dc868776a6feda` | `pretrain.py` +383 lines: same trajectory-augmentation additions. | +| `srm/` | github.com/YurenHao0426/srm (fork of sapientinc/HRM) | tracked separately at that remote | Stable Recursive Model (our architecture). | + +To diff against upstream: clone the upstream repo at the pinned commit and diff `pretrain.py`. + +Excluded from this export (regenerable / heavy): all `checkpoints/`, `wandb/`, `outputs/`, +`__pycache__/`, `*.pt`, `*.npz`, `*.png`, the Sudoku dataset (rebuild via +`scripts/build_datasets.sh sudoku`). The Maze dataset IS included for clone-and-run. |