dataset-curation
Prepare, format, and validate datasets for supervised fine-tuning and preference training. Use when converting raw data into training format, applying chat templates, configuring sequence packing, generating synthetic training data, or writing a dataset card before a run.
By wshobson · 1,441 installs
npx skills add wshobson/agents --skill dataset-curation
Source repository · Upstream listing
Dataset Curation
This skill assumes finetuning method selection
already routed here — the next step is preparing
data, not choosing a method. What follows: format
selection by target method, the template/packing
mechanics behind the most common silent training
failures, rules for mixing in synthetic data
without collapse, and the dataset card that closes
out Phase 2 before a run starts.
Input: raw examples (demonstrations, preference
judgments, or task prompts) plus a routing decision
from finetuning method selection .
Output format: a formatted, packed, validated
JSONL dataset plus a completed dataset card — the
Phase 2 artifact /finetune checks before launching
training.
Format Selection
Method Shape Rows
SFT, single turn Instruct ( instruction / response or prompt / completion ) ~1,000+ floor
SFT, multi turn Conversation / ChatML messages list ~1,000+ floor
DPO / ORPO Preference pair ( prompt , chosen , rejected ) Method dependent, see preference optimization
KTO Unpaired ( prompt , completion , label ) Method dependent, see preference optimization
GRPO / RLVR Prompt only ( prompt + verifier metadata) Method dependent, see grpo rlvr training
~1,000+ rows is the recommended floor for SFT ,
not a target. Below it, a handful of low quality
or duplicate examples can dominate the gradient;
above it, quality over quantity — a smaller
verified, deduplicated set beats a larger noisy one.
The ChatML shape, for orientation; the other four
formats plus a ShareGPT conversion note live in
references/formats and templates.md :
Chat Templates and Loss Masking
Apply the target model's chat template before
any concatenation or packing, never after — packing
raw text and templating the packed blob afterward
corrupts turn boundaries, landing role markers in
the wrong place relative to each example.
Train on assistant responses only. Mask the
loss ( 100 in the labels tensor) over system/user
turns and the template's own role markers — only
assistant turn content tokens contribute to loss.
Template/tokenizer mismatches are a top silent
failure mode. A model trained against one chat
template but served or evaluated with a different
one degrades without erroring. Verify the same
template string used in training is applied at
inference and eval time.
Keep the dataset in messages shape and let
the trainer template and mask it
( assistant only loss=True in current TRL) —
pre rendering to a flat text field destroys the
turn boundaries masking needs. Full code sketch:
references/formats and templates.md . Sanity check
before training — decode only unmasked positions;
expect only assistant text:
Packing
Without packing, 40–70% of compute is spent on
padding — variable length examples batched at a
fixed sequence length waste the gap between each
example's length and the batch's max. Packing
concatenates multiple examples into one sequence
up to the max length, cutting most of that waste.
Packing changes batch semantics. A packed
sequence can contain several original examples, so
"steps per epoch" and any LR schedule keyed to
example count shift once packing is on — recompute
schedule milestones against packed sequence count.
MANDATORY: decode and manually inspect 5–10
packed sequences before scaling to a full run.
Confirm example boundaries land where expected,
template markers are intact per sub example, and
the loss mask is still assistant only within each
packed sequence. Not optional — packing bugs are
silent (the loss curve looks normal) and only
surface in eval quality, hours later:
Synthetic Data Rules
Keep ≥25% real data as a collapse guard.
Training on a growing share of model generated
data without a real data floor drives measurable
quality collapse over successive generations —
25% real is the minimum that holds the line.
General domain replay rows
count toward this floor —
"real" means "not generated
for this task from this
student," not "human authored."
An all synthetic by construction
dataset can meet the ≥25% floor
through replay alone (see
references/synthetic data.md 's
Replay Mix Construction recipe);
state which rows count as "real"
in the dataset card rather than
leaving the floor structurally
unmeetable.
Magpie and rejection sampling are the
workhorses. Magpie extracts prompts from the
model's own template prior; rejection sampling
generates several candidates per prompt and keeps
only the ones a filter passes. Both beat naive
single shot generation.
Targeted, student aware generation beats static
generation by 1.3–2x sample efficiency — aiming
at the student's actual failure modes hits a
quality bar with fewer filtered examples.
Typical accept rates after filtering run
10–30%. Plan volume accordingly — a 10,000 row
target at 15% accept needs ~65,000+ raw generations.
Generation method ranking, filter funnel, replay
mix construction, and distillation pattern:
references/synthetic data.md .
The Dataset Card
Every dataset that reaches training gets a card —
the required Phase 2 artifact /finetune checks
before launching. The card is not free form
documentation; it MUST carry these fields:
Provenance — where every row came from (real
source(s), synthetic method(s), or both),
traceable to trace to training data output.
Counts — total rows, and rows per split
(train/eval/held out) if split.
Synthetic/real ratio — the measured ratio,
checked against the ≥25% real floor above.
Dedup method — exact match, semantic
(embedding threshold), or both; see the filter
funnel in references/synthetic data.md .
Template used — the exact chat template
string/identifier, kept consistent through
inference and eval — this is what ties an
eval harness first run back to the checkpoint.
Packing config — whether packing was used,
max sequence length, and confirmation the
5–10 sequence manual inspection above was done.
A dataset missing any of these six fields isn't
ready for /finetune — the card is a gate, not a
summary written after the fact.
Phase 2 Exit Checklist
Before handing off to /finetune , confirm:
1. Format matches the method (table above).
2. Template applied before concatenation.
3. Loss masked to assistant turns only.
4. 5–10 packed sequences decoded and read.
5. ≥25% real data in the final mix.
6. Dataset card complete — all six fields.
References
references/formats and templates.md — JSONL
examples per format, current TRL masking code,
and the ShareGPT conversion note.
references/synthetic data.md — generation method
ranking, filter funnel, replay mix construction,
and teacher→student distillation pattern.
Related skills: finetuning method selection routes
here; lora qlora recipes , vision sft , and
preference optimization consume the datasets this
skill produces; trace to training data is the
provenance source for graded trajectory datasets;
eval harness first grades the resulting checkpoint.