Mellum2-12B-SFT-DPO-Experiment

An experimental SFT and DPO fine-tune of JetBrains/Mellum2-12B-A2.5B-Thinking using a multi-GPU setup. This model was trained to test distributed training pipelines and data filtering techniques on small datasets. We release 2 model checkpoints. A SFT only trained version and a DPO (with SFT) version.

Pipeline Details

  • SFT Data: 3,000 decontaminated medium/hard samples from nvidia/OpenCodeReasoning.
  • DPO Data: 2,439 sandbox-validated pairs from allenai/Dolci-Think-RL-7B-Completions-DPO.
  • Hardware: 4x NVIDIA A100 (80GB) utilizing PyTorch FSDP full sharding.
  • Precision: BF16

Results & Limitations

When tested against LiveCodeBench v6, the model's Pass@1 score dropped from 0.30 (base) to 0.19. The model appears to have overfitted to the specific reasoning trace formatting of the training data, which introduced syntax styling changes that impacted the evaluation suite.

This model checkpoint is uploaded for infrastructure validation and reproducibility analysis.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for mamoeed/Mellum2-12B-post-post-train

Finetuned
(4)
this model