CoolFace
Modelpublic

Junrulu/Reproduced-tulu2-dpo-13b

sourceHugging Faceupdated 11mo agoView on Hugging Face
1likes23downloads
Model Card

Model Card for Reproduced Tulu2 DPO 13B

This repository provides a reproduction version of Tulu2-DPO-13B finetuned upon Tulu2-13B and Ultrafeedback. Therefore, we obey all licenses mentioned in Tulu2's work. Check our codes for more details: https://github.com/LuJunru/LLM_Finetune/tree/DPO, which is built with TRL.

Performance

ModelSizeAlignmentMT-Bench (score)AlpacaEval 2.0 (win rate %)
Tulu-v2-13b ๐Ÿช13BSFT5.792.61
Tulu-v2-dpo-13b ๐Ÿช13BDPO6.066.96
Reproduced-tulu2-dpo-13b13BDPO6.276.71

Input Format

The model is trained to use the following format (note the newlines):

<|user|>
Your message here!
<|assistant|>

For best results, format all inputs in this manner. Make sure to include a newline after `<|assistant|>`, this can affect generation quality quite a bit. Note: if fine-tuning with this chat template, ensure to evaluate and test with the chat template. Otherwise, fine-tining without the template if you choose to not use template during testing. Any mismatch of the chatting template between training and testing phases can obviously dampen the final performance.

Training hyperparameters

The following hyperparameters were used during DPO training:

  • โ€”DPO beta: 0.1
  • โ€”learning_rate: 1e-6 * sqrt(Num of Nodes)
  • โ€”totaltrainbatch_size: 128 * Num of Nodes
  • โ€”optimizer: AdamW with beta1 0.9, beta2 0.999 and epsilon 1e-8
  • โ€”lrschedulertype: linear
  • โ€”lrschedulerwarmup_ratio: 0.1
  • โ€”Weight Decay: 0.0
  • โ€”num_epochs: 3.0
  • โ€”Specifically add above input format over training samples