gemma-2b-dolly-qa

This model is a fine-tuned version of google/gemma-2b on the generator dataset. It achieves the following results on the evaluation set:

• Loss: 2.0224

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-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_ratio: 0.05
• training_steps: 1480

Training results

Training Loss	Epoch	Step	Validation Loss
2.9192	1.6393	100	2.5716
2.4559	3.2787	200	2.3107
2.2721	4.9180	300	2.1791
2.1671	6.5574	400	2.1127
2.1217	8.1967	500	2.0792
2.0856	9.8361	600	2.0608
2.0752	11.4754	700	2.0495
2.0542	13.1148	800	2.0408
2.0447	14.7541	900	2.0349
2.0428	16.3934	1000	2.0305
2.0303	18.0328	1100	2.0271
2.0207	19.6721	1200	2.0249
2.0237	21.3115	1300	2.0234
2.0191	22.9508	1400	2.0224

Framework versions

• PEFT 0.12.0
• Transformers 4.43.3
• Pytorch 2.1.0.post0+cxx11.abi
• Datasets 2.20.0
• Tokenizers 0.19.1