CoolFace
Modelpublic

daviBera/intern35_2b_lora_expert_counting-102400

sourceHugging Faceotherupdated 7mo agoView on Hugging Face
0likes16downloads
Model Card

<h2 align="center"> Linear Model Merging Unlocks Simple and Scalable Multimodal Data Mixture Optimization

<br>

![arXiv](https://www.arxiv.org/pdf/2602.04937) ![๐Ÿค— Model (HuggingFace)](https://huggingface.co/collections/daviBera/mllms-merging-4-dmo) ![๐Ÿค— Dataset (HuggingFace)](https://huggingface.co/datasets/daviBera/expertsdatasets-102400) [![github](https://img.shields.io/badge/github-repo-blue?logo=github)](https://github.com/BerasiDavide/mLLMsmerging4DMO) </h2>

This is an official checkpoint from the paper: "Linear Model Merging Unlocks Simple and Scalable Multimodal Data Mixture Optimization " (link). See the official implementation for more information on how to use the models.

intern352bloraexpertcounting-102400

This model is a fine-tuned version of OpenGVLab/InternVL3_5-2B-Pretrained-HF on a custom dataset with Counting data (~100k samples).

It achieves the following results on the evaluation set:

  • โ€”Loss: 0.2169

Training hyperparameters

The following hyperparameters were used during training:

  • โ€”learning_rate: 2e-05
  • โ€”trainbatchsize: 8
  • โ€”evalbatchsize: 4
  • โ€”seed: 42
  • โ€”distributed_type: multi-GPU
  • โ€”num_devices: 4
  • โ€”gradientaccumulationsteps: 4
  • โ€”totaltrainbatch_size: 128
  • โ€”totalevalbatch_size: 16
  • โ€”optimizer: Use adamwtorch with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • โ€”lrschedulertype: cosine
  • โ€”lrschedulerwarmup_ratio: 0.1
  • โ€”training_steps: 800

Training results

Training LossEpochStepValidation Loss
0.2740.1251000.2621
0.25390.252000.2416
0.21680.3753000.2279
0.23460.54000.2249
0.22650.6255000.2218
0.21410.756000.2188
0.22780.8757000.2180
0.21711.08000.2169

Framework versions

  • โ€”PEFT 0.15.2
  • โ€”Transformers 4.52.4
  • โ€”Pytorch 2.7.1+cu126
  • โ€”Datasets 3.6.0
  • โ€”Tokenizers 0.21.1