CoolFace
Modelpublic

htlou/backup_0202_llamafactory_RLAIF-V-Dataset-llava-mistral

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

RLAIF-V-Dataset

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

  • —Loss: 0.4513
  • —Rewards/chosen: -3.2808
  • —Rewards/rejected: -6.0928
  • —Rewards/accuracies: 0.8212
  • —Rewards/margins: 2.8121
  • —Logps/rejected: -219.8085
  • —Logps/chosen: -191.2850
  • —Logits/rejected: -2.2605
  • —Logits/chosen: -2.2964

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.59890.1368400.6069-0.3887-0.86150.63650.4728-167.4954-162.3644-2.4012-2.4102
0.54520.2735800.5331-0.8812-1.83380.71350.9526-177.2182-167.2896-2.5177-2.5334
0.50260.41031200.4925-1.4411-2.67030.74421.2292-185.5836-172.8887-1.9765-2.0268
0.45110.54701600.4683-1.3283-3.02840.76251.7001-189.1644-171.7603-2.0280-2.0709
0.45620.68382000.4528-1.4943-3.26750.75671.7732-191.5553-173.4200-2.1029-2.1462
0.41890.82052400.4494-1.9309-3.88990.76631.9589-197.7792-177.7867-2.4165-2.4472
0.44840.95732800.4432-1.7397-3.82380.76352.0841-197.1187-175.8746-2.1586-2.2000
0.2221.09403200.4504-1.2207-2.96980.77601.7491-188.5780-170.6839-2.4060-2.4397
0.20181.23083600.4438-2.0855-4.47460.78852.3891-203.6262-179.3325-2.3445-2.3790
0.20171.36754000.4350-1.9109-4.14140.79812.2305-200.2943-177.5862-2.3022-2.3351
0.19991.50434400.4288-2.1056-4.46410.80482.3585-203.5214-179.5331-2.1361-2.1716
0.18371.64104800.4262-2.2318-4.70560.81252.4738-205.9359-180.7949-2.2127-2.2452
0.19421.77785200.4163-2.3806-5.02830.81152.6478-209.1637-182.2829-2.3333-2.3675
0.18211.91455600.4165-2.2038-4.67090.81732.4671-205.5893-180.5155-2.3238-2.3543
0.08582.05136000.4415-2.7029-5.19790.81442.4950-210.8597-185.5066-2.2872-2.3220
0.08322.18806400.4414-2.8951-5.65540.81732.7603-215.4344-187.4282-2.2892-2.3247
0.08172.32486800.4521-3.2403-6.00140.81542.7611-218.8945-190.8804-2.2697-2.3056
0.08582.46157200.4479-3.3847-6.30120.82212.9165-221.8926-192.3248-2.2708-2.3072
0.07232.59837600.4574-3.3436-6.11130.81732.7677-219.9932-191.9133-2.2754-2.3103
0.07172.73508000.4532-3.3171-6.12890.81922.8118-220.1688-191.6483-2.2610-2.2973
0.06912.87188400.4514-3.2739-6.08550.82122.8116-219.7354-191.2166-2.2604-2.2964

Framework versions

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