CoolFace
Modelpublic

argilla/zephyr-7b-spin-iter3-v0

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
8likes25downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

zephyr-7b-spin-iter3-v0

A model matching the results of SPIN with very little data (30x less), carefully curated by the amazing Data Is Better Together community

<div> <img src="https://cdn-uploads.huggingface.co/production/uploads/60420dccc15e823a685f2b03/aEzpD6gvn0xOrN2rNzpZI.webp"> </div>

<p align="center"> <a href="https://github.com/argilla-io/distilabel"> <img src="https://raw.githubusercontent.com/argilla-io/distilabel/main/docs/assets/distilabel-badge-light.png" alt="Built with Distilabel" width="200" height="32"/> </a> </p>

This model is a fine-tuned version of argilla/zephyr-7b-spin-iter2-v0 on the argilla/10k_prompts_SPIN_iter3_zephyr_top and the argilla/10k_prompts_SPIN_iter2_zephyr_top dataset.

Check this repo for full reproducible code using the original SPIN implementation and distilabel.

If you want to contribute to high quality datasets like this, contribute to the DIBT prompt collective initiative.

MT-Bench results

Model1st Turn Score2nd Turn ScoreAverage ScoreSPIN paper Score
zephyr-7b-sft-full6.66256.02506.343755.94
zephyr-7b-spin-iter0-v06.643756.17506.4093756.46
zephyr-7b-spin-iter1-v06.906256.30006.6031256.65
zephyr-7b-spin-iter2-v07.13756.31256.7250006.78
zephyr-7b-spin-iter3-v07.093756.45006.771875-

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • —learning_rate: 1e-07
  • —trainbatchsize: 8
  • —evalbatchsize: 8
  • —seed: 42
  • —distributed_type: multi-GPU
  • —num_devices: 4
  • —gradientaccumulationsteps: 2
  • —totaltrainbatch_size: 64
  • —totalevalbatch_size: 32
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —lrschedulerwarmup_ratio: 0.1
  • —num_epochs: 2.0

Training results

Training LossEpochStepValidation LossRewards/realRewards/generatedRewards/accuraciesRewards/marginsLogps/generatedLogps/realLogits/generatedLogits/real
0.29280.49250.3951-2.6212-20.32680.906217.7056-700.5638-278.0876-2.8098-2.8090
0.14870.97500.1319-2.9077-29.14590.937526.2382-702.3276-278.1449-2.8218-2.8066
0.0061.46750.1269-2.6037-29.15190.958326.5482-702.3289-278.0841-2.8175-2.8037
0.00861.941000.1099-2.9181-29.69700.927126.7789-702.4378-278.1470-2.8177-2.8051

Framework versions

  • —Transformers 4.37.0
  • —Pytorch 2.1.2+cu121
  • —Datasets 2.14.6
  • —Tokenizers 0.15.2