Files

T

张宗平 bf95ac18e4 feat(m3): stage 1 alignment training script + smoke (T3.2)

- src/tsmm/train/stage1.py: streams align.jsonl, freeze_llm, trains
  Encoder+Projector (AdamW lr=1e-4), bs=8/grad_accum=4/ctx=512/BF16,
  gradient checkpointing + enable_input_require_grads, loss = lm + lambda*InfoNCE
  (original vs perturbed TS representation), ckpt + tensorboard every 2000 steps.
- scripts/train_stage1.sh: wrapper with design defaults.
- src/tsmm/data/collator.py: handle JSON null (missing markers) -> NaN -> fill.
- .gitignore: /checkpoints/ and data/*.jsonl.
- Smoke verified: --max_steps 5 on 64 samples, finite loss, peak 5.52 GB (≤6GB).
- 100 tests passing.

2026-06-30 02:22:32 +00:00

gen_synthetic.py

feat(m1): offline generation pipeline + CLI (T1.5)

2026-06-29 23:29:04 +08:00

README.md

feat(m1): scaffold tsmm package (T1.1)

2026-06-29 23:21:47 +08:00

train_stage1.sh

feat(m3): stage 1 alignment training script + smoke (T3.2)

2026-06-30 02:22:32 +00:00

README.md

Scripts directory

Offline entrypoints: gen_synthetic.py, train_stage1.sh, train_stage2.sh, run_eval.sh.