Downloads · 30 days
0
pengxiang/fprm-flagship-L4
fprm-flagship-L4 is a machine learning model from pengxiang. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Raw research checkpoints for two flagship runs. Accuracy numbers are not included here on purpose: they are re-measured with the frozen eval pipeline.
Downloads · 30 days
0
Access
Public
Updated Aug 4, 2026
Repo size
10.7 GB
Likes
0
Public
Click a slice to open those files.
.pt6.6 GB · 61%
From the Hugging Face model README
Raw research checkpoints for two flagship runs. Accuracy numbers are not included here on purpose: they are re-measured with the frozen eval pipeline.
| Run | loop_attnres | Final step | Size |
|---|---|---|---|
fl-b01-L4-lrdecay-s1 | true (grid=ema, ema_heads=1, beta_init=0.1, mode=state, window=16) | 78120 / 78125 | 4.2 GB |
fl-carrylast-L4-s1 | false | 78120 / 78125 | 5.8 GB |
trm.pyThe two arms were trained with different trm.py revisions (differing md5).
fl-carrylast-L4-s1's behavioural difference lives in its code, not in its
config or parameters — its state dict has no loop_attn.* params at any step
(verified at 5000 / 45570 / 78120, so no mid-run architecture change).
Each directory ships the exact trm.py, losses.py and all_config.yaml it
was trained with. Do not cross-load.
data/sudoku-extreme-1k-aug-1000, hidden 512, L_layers=4, H_cycles=3,
L_cycles=6, halt_max_steps=16, rope, bf16, stablemax_cross_entropy,
q_loss_coeff=0.5. Full config in each all_config.yaml.
step_<N> — model state dict only (109 MB)step_<N>_train_state.pt — model + optimizer, for resuming (273 MB)all_config.yaml, trm.py, losses.pylogs/)Both runs reached 78,125 steps and ended exit=0 / DONE. No NaN and no OOM
in either log. Both were restarted and resumed from checkpoints mid-training.
The b01 log contains 6 tracebacks, all infrastructure-level, none a training
failure: 4x ModuleNotFoundError: coolname (launch-env, fast-failed and
relaunched) and a disk-full (Errno 28) while writing step_65100 on
2026-07-17, after which the run resumed and completed.
Last logged train metrics (not eval):
| Run | lm_loss | seq_acc | q_halt_acc |
|---|---|---|---|
fl-b01-L4-lrdecay-s1 | 0.63375 | 0.8547 | 0.98291 |
fl-carrylast-L4-s1 | 0.64277 | 0.85833 | 1.0 |