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 /trm/requirements.txt | |
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 'trm/requirements.txt')
| -rw-r--r-- | trm/requirements.txt | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/trm/requirements.txt b/trm/requirements.txt new file mode 100644 index 0000000..cea0f55 --- /dev/null +++ b/trm/requirements.txt @@ -0,0 +1,20 @@ +torch +adam-atan2 +einops +tqdm +coolname +pydantic +argdantic +wandb +omegaconf +hydra-core +huggingface_hub +packaging +ninja +wheel +setuptools +setuptools-scm +pydantic-core +huggingface_hub +numba +triton
\ No newline at end of file |