Model save

README.md

@@ -2,13 +2,12 @@
 license: apache-2.0
 library_name: peft
 tags:
-- alignment-handbook
 - trl
 - sft
 - generated_from_trainer
 base_model: mistralai/Mistral-7B-v0.1
 datasets:
-- HuggingFaceH4/ultrachat_200k
+- generator
 model-index:
 - name: zephyr-7b-sft-qlora
   results: []
@@ -19,7 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # zephyr-7b-sft-qlora
 
-This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on the HuggingFaceH4/ultrachat_200k dataset.
+This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on the generator dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.9584
 
 ## Model description
 
@@ -40,20 +41,23 @@ More information needed
 The following hyperparameters were used during training:
 - learning_rate: 0.0002
 - train_batch_size: 2
-- eval_batch_size: 4
+- eval_batch_size: 2
 - seed: 42
 - distributed_type: multi-GPU
 - num_devices: 4
 - gradient_accumulation_steps: 4
 - total_train_batch_size: 32
-- total_eval_batch_size: 16
+- total_eval_batch_size: 8
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 2
+- num_epochs: 1
 
 ### Training results
 
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 0.98          | 1.0   | 870  | 0.9584          |
 
 
 ### Framework versions

adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8303e4d990e92e274c57096f600cbff3830f053daf12c7c229ac7308334634c2
+oid sha256:8a8495a8724d32fa7528dafb12429bd1e044b1dca6ac8f7a6d14f160b6c5113e
 size 83946192

all_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 0.17,
-    "train_loss": 0.0,
-    "train_runtime": 1.4376,
+    "epoch": 1.0,
+    "train_loss": 0.9710182381772446,
+    "train_runtime": 17696.8906,
     "train_samples": 41573,
-    "train_samples_per_second": 38742.113,
-    "train_steps_per_second": 1210.343
+    "train_samples_per_second": 1.574,
+    "train_steps_per_second": 0.049
 }

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 0.17,
-    "train_loss": 0.0,
-    "train_runtime": 1.4376,
+    "epoch": 1.0,
+    "train_loss": 0.9710182381772446,
+    "train_runtime": 17696.8906,
     "train_samples": 41573,
-    "train_samples_per_second": 38742.113,
-    "train_steps_per_second": 1210.343
+    "train_samples_per_second": 1.574,
+    "train_steps_per_second": 0.049
 }