Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: other
|
| 3 |
+
tags: [meta-reasoning, qwen3_5, condgen]
|
| 4 |
+
---
|
| 5 |
+
|
| 6 |
+
# MR_midtrain_9B_v5_condgen
|
| 7 |
+
|
| 8 |
+
This is the **final step-4128 checkpoint**, not the best-validation checkpoint,
|
| 9 |
+
of a six-epoch V5 supervised midtraining run from the stock Qwen3.5-9B weights.
|
| 10 |
+
The model uses the `Qwen3_5ForConditionalGeneration` architecture and is
|
| 11 |
+
published with the vision tower in the flat `model.visual.*` key layout expected
|
| 12 |
+
by verl's Megatron bridge. The export changes only the names of 333 vision
|
| 13 |
+
tensors from `model.language_model.visual.*`; their values are not modified.
|
| 14 |
+
The vision tower is carried from the base model and was not a target of the
|
| 15 |
+
text-only SFT run.
|
| 16 |
+
|
| 17 |
+
## Training data and format
|
| 18 |
+
|
| 19 |
+
The training split contained 88,115 examples: 49,188 exploration (E), 29,450
|
| 20 |
+
meta-reasoning (MR), 6,477 final-answer (FA), and 3,000 termination traces.
|
| 21 |
+
It was derived from `HerrHruby/MR_midtrain_V5_sft` with additional termination
|
| 22 |
+
traces sampled from existing frontiers. The unchanged validation split had 258
|
| 23 |
+
examples. Training used the V4 verl/FSDP recipe with a 65,536-token maximum
|
| 24 |
+
training length and the V5 attribution-aware tokenizer.
|
| 25 |
+
|
| 26 |
+
FA training targets retain the FA chain of thought through `</think>`, then emit
|
| 27 |
+
`<attribution>...</attribution>` before the final answer or proof. The tokenizer
|
| 28 |
+
preserves the V4 `<direction>`, `</direction>`, `<summary>`, and `</summary>`
|
| 29 |
+
tokens (IDs 248077–248080) and adds `<attribution>` and `</attribution>` as
|
| 30 |
+
atomic special tokens (IDs 248081–248082). The model's embedding size is
|
| 31 |
+
unchanged.
|
| 32 |
+
|
| 33 |
+
The final checkpoint's validation loss was 0.642673. The lowest validation
|
| 34 |
+
loss in the run was 0.642636 at step 2700, but that checkpoint is **not** the
|
| 35 |
+
one published here. These losses are training diagnostics, not a downstream
|
| 36 |
+
benchmark evaluation.
|
| 37 |
+
|
| 38 |
+
## Compatibility
|
| 39 |
+
|
| 40 |
+
The flat vision-tower key layout matches the published
|
| 41 |
+
[`MR_midtrain_9B_v4_condgen`](https://huggingface.co/HerrHruby/MR_midtrain_9B_v4_condgen)
|
| 42 |
+
layout and is intended for verl/Megatron loading. It also retains the full
|
| 43 |
+
conditional-generation architecture for Transformers and vLLM. For text-only
|
| 44 |
+
vLLM serving, disable image and video inputs (for example with
|
| 45 |
+
`--limit-mm-per-prompt '{"image":0,"video":0}'`).
|
| 46 |
+
|
| 47 |
+
The export process first merges the final verl FSDP model shards into a
|
| 48 |
+
Hugging Face safetensors checkpoint, then re-keys the 333 vision tensors.
|
| 49 |
+
Optimizer and dataloader states are not part of this inference/RL-init export.
|