From 66e0d8b9fd4d0f7a2231d689c055e26fdf1cf04a Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Sat, 13 Jun 2026 12:35:36 -0500 Subject: rrm workspace: TRM/HRM/SRM code, Maze dataset, dynamical-analysis pipeline 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 --- PROVENANCE.md | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 PROVENANCE.md (limited to 'PROVENANCE.md') 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. -- cgit v1.2.3