--- license: apache-2.0 library_name: peft tags: - trl - sft - generated_from_trainer base_model: mistralai/Mistral-7B-Instruct-v0.2 datasets: - generator model-index: - name: mistral-finetune results: [] --- # mistral-finetune This model is a fine-tuned version of [mistralai/Mistral-7B-Instruct-v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2) on the generator dataset. It achieves the following results on the evaluation set: - Loss: 0.3252 ## 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: 2.5e-05 - train_batch_size: 4 - eval_batch_size: 8 - seed: 42 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08 - lr_scheduler_type: linear - lr_scheduler_warmup_steps: 0.03 - num_epochs: 5 - mixed_precision_training: Native AMP ### Training results | Training Loss | Epoch | Step | Validation Loss | |:-------------:|:-----:|:----:|:---------------:| | 0.3784 | 1.0 | 201 | 0.3443 | | 0.3315 | 2.0 | 402 | 0.3313 | | 0.3043 | 3.0 | 603 | 0.3252 | | 0.3079 | 4.0 | 804 | 0.3236 | | 0.2966 | 5.0 | 1005 | 0.3252 | ### Framework versions - PEFT 0.10.1.dev0 - Transformers 4.38.2 - Pytorch 2.1.1+cu121 - Datasets 2.18.0 - Tokenizers 0.15.2