ncis / README.md
BaoTNguyen's picture
baotnguyen/ncis
0e14043 verified
|
raw
history blame
2.16 kB
metadata
base_model: mistralai/Mistral-7B-v0.1
library_name: peft
license: apache-2.0
metrics:
  - accuracy
tags:
  - generated_from_trainer
model-index:
  - name: ncis
    results: []

Visualize in Weights & Biases

ncis

This model is a fine-tuned version of mistralai/Mistral-7B-v0.1 on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 1.5884
  • Accuracy: 0.8205

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.0001
  • train_batch_size: 16
  • eval_batch_size: 16
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 10

Training results

Training Loss Epoch Step Validation Loss Accuracy
No log 1.0 19 4.9428 0.4786
No log 2.0 38 4.6067 0.6325
No log 3.0 57 1.0003 0.8632
No log 4.0 76 1.4983 0.8376
No log 5.0 95 1.5749 0.8291
No log 6.0 114 1.5852 0.8205
No log 7.0 133 1.5872 0.8205
No log 8.0 152 1.5881 0.8205
No log 9.0 171 1.5883 0.8205
No log 10.0 190 1.5884 0.8205

Framework versions

  • PEFT 0.11.1
  • Transformers 4.42.4
  • Pytorch 2.1.2
  • Datasets 2.16.0
  • Tokenizers 0.19.1