metadata
license: other
library_name: peft
tags:
- trl
- reward-trainer
- generated_from_trainer
base_model: google/gemma-2b
metrics:
- accuracy
model-index:
- name: gemma_2b_social_reasoning_reward_model
results: []
gemma_2b_social_reasoning_reward_model
This model is a fine-tuned version of google/gemma-2b on an unknown dataset. It achieves the following results on the evaluation set:
- Loss: 0.6045
- Accuracy: 0.6818
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: 0.0005
- train_batch_size: 16
- eval_batch_size: 16
- seed: 42
- gradient_accumulation_steps: 4
- total_train_batch_size: 64
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: cosine
- lr_scheduler_warmup_steps: 10
- num_epochs: 3
Training results
Training Loss | Epoch | Step | Validation Loss | Accuracy |
---|---|---|---|---|
0.8047 | 0.24 | 10 | 0.7041 | 0.5447 |
0.6758 | 0.48 | 20 | 0.6179 | 0.6725 |
0.6436 | 0.72 | 30 | 0.6066 | 0.6690 |
0.6423 | 0.96 | 40 | 0.6053 | 0.6813 |
0.5838 | 1.2 | 50 | 0.6080 | 0.6637 |
0.5871 | 1.44 | 60 | 0.6301 | 0.6480 |
0.5928 | 1.68 | 70 | 0.6338 | 0.6515 |
0.5777 | 1.92 | 80 | 0.6198 | 0.6375 |
0.5378 | 2.16 | 90 | 0.6225 | 0.6392 |
0.5163 | 2.4 | 100 | 0.6222 | 0.6532 |
0.4963 | 2.63 | 110 | 0.6224 | 0.6567 |
0.5176 | 2.87 | 120 | 0.6229 | 0.6550 |
Framework versions
- PEFT 0.9.0
- Transformers 4.38.2
- Pytorch 2.2.1+cu121
- Datasets 2.18.0
- Tokenizers 0.15.2