CoolFace
Modelpublic

htlou/backup_0202_llamafactory_AA_preference_l0_new_step10_0_60-llava-mistral

sourceHugging Faceotherupdated 2y agoView on Hugging Face
0likes5downloads
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. -->

AApreferencel0newstep10060

This model is a fine-tuned version of llava-hf/llava-v1.6-mistral-7b-hf on the AApreferencel0newstep10060 dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.5930
  • —Rewards/chosen: 0.5295
  • —Rewards/rejected: -1.8896
  • —Rewards/accuracies: 0.7917
  • —Rewards/margins: 2.4190
  • —Logps/rejected: -242.2141
  • —Logps/chosen: -237.5954
  • —Logits/rejected: -2.0873
  • —Logits/chosen: -2.1246

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-06
  • —trainbatchsize: 8
  • —evalbatchsize: 8
  • —seed: 42
  • —distributed_type: multi-GPU
  • —num_devices: 8
  • —gradientaccumulationsteps: 4
  • —totaltrainbatch_size: 256
  • —totalevalbatch_size: 64
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: cosine
  • —lrschedulerwarmup_steps: 10
  • —num_epochs: 3.0

Training results

Training LossEpochStepValidation LossRewards/chosenRewards/rejectedRewards/accuraciesRewards/marginsLogps/rejectedLogps/chosenLogits/rejectedLogits/chosen
0.57280.6231500.59680.7303-0.46940.72571.1997-228.0123-235.5866-2.3844-2.3915
0.23381.24611000.61380.8569-1.07170.77081.9286-234.0351-234.3208-2.3760-2.3908
0.26531.86921500.58300.4703-1.73330.78472.2035-240.6510-238.1874-2.0378-2.0764
0.15962.49222000.59090.5892-1.80170.79512.3909-241.3353-236.9982-2.0980-2.1340

Framework versions

  • —Transformers 4.45.2
  • —Pytorch 2.4.0+cu121
  • —Datasets 2.21.0
  • —Tokenizers 0.20.3