paligemma-cnmc-ft / README.md
dwb2023's picture
dwb2023/paligemma-cnmc-ft
36fc508 verified
|
raw
history blame
1.81 kB
metadata
base_model: google/paligemma-3b-pt-224
library_name: peft
license: gemma
tags:
  - generated_from_trainer
model-index:
  - name: paligemma-cnmc-ft
    results: []

paligemma-cnmc-ft

This model is a fine-tuned version of google/paligemma-3b-pt-224 on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.3006

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: 2e-05
  • train_batch_size: 2
  • eval_batch_size: 8
  • seed: 42
  • gradient_accumulation_steps: 8
  • total_train_batch_size: 16
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • lr_scheduler_warmup_steps: 50
  • num_epochs: 100

Training results

Training Loss Epoch Step Validation Loss
No log 0.9645 17 1.2278
No log 1.9858 35 0.4162
0.8676 2.9504 52 0.3132
0.8676 3.9716 70 0.2602
0.8676 4.9929 88 0.2446
0.2526 5.9574 105 0.2100
0.2526 6.9787 123 0.1986
0.2526 8.0 141 0.3006

Framework versions

  • PEFT 0.11.1
  • Transformers 4.43.0.dev0
  • Pytorch 2.3.0+cu121
  • Datasets 2.20.0
  • Tokenizers 0.19.1