deadcode99/mistral-billm-token-classification-segmentwise

README.md

@@ -21,11 +21,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [mistralai/Mistral-7B-Instruct-v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.3392
+- Loss: 0.4998
 - Precision: 0.0
 - Recall: 0.0
 - F1: 0.0
-- Accuracy: 0.7072
+- Accuracy: 0.7829
 
 ## Model description
 
@@ -44,7 +44,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0001
+- learning_rate: 0.001
 - train_batch_size: 1
 - eval_batch_size: 1
 - seed: 42
@@ -52,16 +52,22 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 32
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 4
+- num_epochs: 10
 
 ### Training results
 
 | Training Loss | Epoch  | Step | Validation Loss | Precision | Recall | F1  | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:---------:|:------:|:---:|:--------:|
-| No log        | 0.9784 | 34   | 1.5589          | 0.0       | 0.0    | 0.0 | 0.6931   |
-| No log        | 1.9856 | 69   | 1.3685          | 0.0       | 0.0    | 0.0 | 0.7023   |
-| No log        | 2.9928 | 104  | 1.3357          | 0.0       | 0.0    | 0.0 | 0.7067   |
-| No log        | 3.9137 | 136  | 1.3392          | 0.0       | 0.0    | 0.0 | 0.7072   |
+| No log        | 0.9784 | 34   | 0.9557          | 0.0       | 0.0    | 0.0 | 0.7596   |
+| No log        | 1.9856 | 69   | 0.7691          | 0.0       | 0.0    | 0.0 | 0.7707   |
+| No log        | 2.9928 | 104  | 0.7086          | 0.0       | 0.0    | 0.0 | 0.7794   |
+| No log        | 4.0    | 139  | 0.5693          | 0.0       | 0.0    | 0.0 | 0.7697   |
+| No log        | 4.9784 | 173  | 0.5449          | 0.0       | 0.0    | 0.0 | 0.7758   |
+| No log        | 5.9856 | 208  | 0.5168          | 0.0       | 0.0    | 0.0 | 0.7805   |
+| No log        | 6.9928 | 243  | 0.5379          | 0.0       | 0.0    | 0.0 | 0.7838   |
+| No log        | 8.0    | 278  | 0.5301          | 0.0       | 0.0    | 0.0 | 0.7847   |
+| No log        | 8.9784 | 312  | 0.5007          | 0.0       | 0.0    | 0.0 | 0.7829   |
+| No log        | 9.7842 | 340  | 0.4998          | 0.0       | 0.0    | 0.0 | 0.7829   |
 
 
 ### Framework versions

adapter_config.json

@@ -20,8 +20,8 @@
   "rank_pattern": {},
   "revision": null,
   "target_modules": [
-    "v_proj",
-    "q_proj"
+    "q_proj",
+    "v_proj"
   ],
   "task_type": "TOKEN_CLS",
   "use_dora": false,

adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4c2f9fc24916cfcecbb9249dfeb3691decc59a26b7ebb4db8eb9ba6749cbf5b9
+oid sha256:761c651de2687afee0d6819c465cac47ab39e50fbd77e288501079f8fc651276
 size 27296852

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a540de03a8af24d3e7aa04e434c5262ede6e5a396afd1dd32c23e76faa70931e
+oid sha256:c09d3d1461ad4afa9aa8380d1ce2b1244989e1eba46cdcab4347a61033b62ac4
 size 5176