Instructions to use NeelContractor/finetuning-checkpoints with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use NeelContractor/finetuning-checkpoints with PEFT:
Task type is invalid.
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Unsloth Desktop
Model card: update GRPO eval with reasoning warm-up results
Browse files
README.md
CHANGED
|
@@ -57,7 +57,7 @@ not from the DPO adapter.
|
|
| 57 |
| `models/sft_qwen06/final` + `checkpoint-100/150/198` | LoRA adapter | Instruction SFT | `cpt_qwen06_merged` |
|
| 58 |
| `models/sft_qwen06_merged` | Full model (fp) | merged SFT | Qwen3-0.6B |
|
| 59 |
| `models/dpo_qwen06/final` + `checkpoint-50/77` | LoRA adapter (+ `ref/`) | DPO preference tuning | `sft_qwen06_merged` |
|
| 60 |
-
| `models/reasoning_warmup_qwen06/final` + `checkpoint-
|
| 61 |
| `models/reasoning_warmup_qwen06_merged` | Full model (fp) | merged warm-up | Qwen3-0.6B |
|
| 62 |
| `models/grpo_qwen06/llm` | LoRA adapter | GRPO (latest) | `reasoning_warmup_qwen06_merged` |
|
| 63 |
| `models/grpo_qwen06/llm_best_0` | LoRA adapter | GRPO (best eval reward) | `reasoning_warmup_qwen06_merged` |
|
|
@@ -154,6 +154,8 @@ print(tokenizer.decode(
|
|
| 154 |
| Hardware | Kaggle, 2× Tesla T4 (15 GB), DDP |
|
| 155 |
| Precision | fp16 training on T4; merged checkpoints saved as bf16 |
|
| 156 |
| Max sequence length | 1024 (SFT / reasoning / GRPO) |
|
|
|
|
|
|
|
| 157 |
|
| 158 |
### Data
|
| 159 |
|
|
@@ -173,20 +175,22 @@ From `models/grpo_qwen06/logs/eval_summary_5.json` (64 held-out prompts):
|
|
| 173 |
|
| 174 |
| Metric | Mean |
|
| 175 |
|---|---|
|
| 176 |
-
| `total_reward` | **
|
| 177 |
-
| `neuraltxt_reward` | **0.
|
| 178 |
-
| `length_penalty_reward` | −0.
|
| 179 |
-
| `num_tokens` |
|
| 180 |
-
| `think_format_reward` | 0.
|
| 181 |
-
| `output_datatype_reward` | 0.
|
| 182 |
-
| `output_schema_reward` | 0.
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
|
| 188 |
-
|
| 189 |
-
|
|
|
|
|
|
|
| 190 |
|
| 191 |
---
|
| 192 |
|
|
@@ -208,7 +212,9 @@ python cloud_train.py --stages reasoning grpo --num-gpus 2
|
|
| 208 |
```
|
| 209 |
|
| 210 |
See the [project README](https://github.com/NeelContractor/finetuning_recipes)
|
| 211 |
-
for stage-by-stage details, dataset prep, and evaluation scripts.
|
|
|
|
|
|
|
| 212 |
|
| 213 |
---
|
| 214 |
|
|
@@ -217,6 +223,7 @@ for stage-by-stage details, dataset prep, and evaluation scripts.
|
|
| 217 |
- Research / educational checkpoints from a short, small-model run.
|
| 218 |
- Not instruction-tuned for production, not safety-tuned, not evaluated for
|
| 219 |
harmful content.
|
| 220 |
-
-
|
|
|
|
| 221 |
- Derived from Qwen3-0.6B (Apache-2.0); check the licenses of the training
|
| 222 |
datasets and the reward model before redistributing or commercial use.
|
|
|
|
| 57 |
| `models/sft_qwen06/final` + `checkpoint-100/150/198` | LoRA adapter | Instruction SFT | `cpt_qwen06_merged` |
|
| 58 |
| `models/sft_qwen06_merged` | Full model (fp) | merged SFT | Qwen3-0.6B |
|
| 59 |
| `models/dpo_qwen06/final` + `checkpoint-50/77` | LoRA adapter (+ `ref/`) | DPO preference tuning | `sft_qwen06_merged` |
|
| 60 |
+
| `models/reasoning_warmup_qwen06/final` + `checkpoint-250/300/324` | LoRA adapter | Reasoning-format SFT (`think`/answer) | `sft_qwen06_merged` |
|
| 61 |
| `models/reasoning_warmup_qwen06_merged` | Full model (fp) | merged warm-up | Qwen3-0.6B |
|
| 62 |
| `models/grpo_qwen06/llm` | LoRA adapter | GRPO (latest) | `reasoning_warmup_qwen06_merged` |
|
| 63 |
| `models/grpo_qwen06/llm_best_0` | LoRA adapter | GRPO (best eval reward) | `reasoning_warmup_qwen06_merged` |
|
|
|
|
| 154 |
| Hardware | Kaggle, 2× Tesla T4 (15 GB), DDP |
|
| 155 |
| Precision | fp16 training on T4; merged checkpoints saved as bf16 |
|
| 156 |
| Max sequence length | 1024 (SFT / reasoning / GRPO) |
|
| 157 |
+
| Reasoning warm-up | 10,000 sampled → 6,911 after the 1024-token filter; 3 epochs, 324 steps, ~46 min on 2×T4 |
|
| 158 |
+
| GRPO | 1 epoch, 8 train events (384 optimizer steps), ~50 min on 2×T4 |
|
| 159 |
|
| 160 |
### Data
|
| 161 |
|
|
|
|
| 175 |
|
| 176 |
| Metric | Mean |
|
| 177 |
|---|---|
|
| 178 |
+
| `total_reward` | **2.355** |
|
| 179 |
+
| `neuraltxt_reward` | **0.519** |
|
| 180 |
+
| `length_penalty_reward` | −0.078 |
|
| 181 |
+
| `num_tokens` | 190.6 |
|
| 182 |
+
| `think_format_reward` | **0.953** |
|
| 183 |
+
| `output_datatype_reward` | 0.461 |
|
| 184 |
+
| `output_schema_reward` | 0.461 |
|
| 185 |
+
|
| 186 |
+
Datatype and schema match rate: **93.8%** each. Best inference score: **2.355**.
|
| 187 |
+
|
| 188 |
+
An earlier run used a reasoning warm-up of only 3 optimizer steps, so the model
|
| 189 |
+
never emitted `think...` blocks and every format reward stayed at zero. The
|
| 190 |
+
current checkpoints come from the dedicated `reasoning` stage (10,000 rows,
|
| 191 |
+
3 epochs), which fixed it: 95.3% of eval responses contain a valid
|
| 192 |
+
`think...` block and 93.8% match the requested datatype/schema. Semantic
|
| 193 |
+
quality is still modest — this is a 0.6B research model, not a strong reasoner.
|
| 194 |
|
| 195 |
---
|
| 196 |
|
|
|
|
| 212 |
```
|
| 213 |
|
| 214 |
See the [project README](https://github.com/NeelContractor/finetuning_recipes)
|
| 215 |
+
for stage-by-stage details, dataset prep, and evaluation scripts. The
|
| 216 |
+
Kaggle runbook for the reasoning branch is
|
| 217 |
+
[REASON_TRAINING.md](https://github.com/NeelContractor/finetuning_recipes/blob/main/REASON_TRAINING.md).
|
| 218 |
|
| 219 |
---
|
| 220 |
|
|
|
|
| 223 |
- Research / educational checkpoints from a short, small-model run.
|
| 224 |
- Not instruction-tuned for production, not safety-tuned, not evaluated for
|
| 225 |
harmful content.
|
| 226 |
+
- Format compliance is now high (95.3% valid think blocks, 93.8% structural
|
| 227 |
+
match), but semantic reasoning quality is limited by the 0.6B model.
|
| 228 |
- Derived from Qwen3-0.6B (Apache-2.0); check the licenses of the training
|
| 229 |
datasets and the reward model before redistributing or commercial use.
|