End of training

README.md

@@ -1,14 +1,14 @@
 ---
 library_name: peft
 license: apache-2.0
-base_model: EleutherAI/pythia-160m
+base_model: EleutherAI/pythia-410m-deduped
 tags:
 - axolotl
 - generated_from_trainer
 datasets:
-- Aivesa/dataset_71c4d505-b734-4315-b4e8-1ed979528a1d
+- Aivesa/dataset_8b4ee2ef-6f34-478c-b009-c74daaa5693c
 model-index:
-- name: bbafd374-4aae-4af5-b2e8-6cd470f8d89f
+- name: b0732ad0-1309-4d92-a6b8-78235a5a5912
   results: []
 ---
 
@@ -21,18 +21,17 @@ should probably proofread and complete it, then remove this comment. -->
 axolotl version: `0.6.0`
 ```yaml
 adapter: lora
-base_model: EleutherAI/pythia-160m
+base_model: EleutherAI/pythia-410m-deduped
 bf16: auto
 chat_template: llama3
 dataset_prepared_path: /workspace/axolotl/data/prepared
 datasets:
 - ds_type: json
   format: custom
-  path: Aivesa/dataset_71c4d505-b734-4315-b4e8-1ed979528a1d
+  path: Aivesa/dataset_8b4ee2ef-6f34-478c-b009-c74daaa5693c
   type:
-    field_input: prompt_setting
-    field_instruction: prompt
-    field_output: completion
+    field_instruction: message_1
+    field_output: message_2
     system_format: '{system}'
     system_prompt: ''
 debug: null
@@ -48,7 +47,7 @@ fsdp_config: null
 gradient_accumulation_steps: 4
 gradient_checkpointing: false
 group_by_length: false
-hub_model_id: Aivesa/bbafd374-4aae-4af5-b2e8-6cd470f8d89f
+hub_model_id: Aivesa/b0732ad0-1309-4d92-a6b8-78235a5a5912
 hub_private_repo: true
 hub_repo: null
 hub_strategy: checkpoint
@@ -90,10 +89,10 @@ use_accelerate: true
 val_set_size: 0.05
 wandb_entity: null
 wandb_mode: online
-wandb_name: 71c4d505-b734-4315-b4e8-1ed979528a1d
+wandb_name: 8b4ee2ef-6f34-478c-b009-c74daaa5693c
 wandb_project: Gradients-On-Demand
 wandb_run: your_name
-wandb_runid: 71c4d505-b734-4315-b4e8-1ed979528a1d
+wandb_runid: 8b4ee2ef-6f34-478c-b009-c74daaa5693c
 warmup_steps: 10
 weight_decay: 0.0
 xformers_attention: null
@@ -102,11 +101,11 @@ xformers_attention: null
 
 </details><br>
 
-# bbafd374-4aae-4af5-b2e8-6cd470f8d89f
+# b0732ad0-1309-4d92-a6b8-78235a5a5912
 
-This model is a fine-tuned version of [EleutherAI/pythia-160m](https://huggingface.co/EleutherAI/pythia-160m) on the Aivesa/dataset_71c4d505-b734-4315-b4e8-1ed979528a1d dataset.
+This model is a fine-tuned version of [EleutherAI/pythia-410m-deduped](https://huggingface.co/EleutherAI/pythia-410m-deduped) on the Aivesa/dataset_8b4ee2ef-6f34-478c-b009-c74daaa5693c dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.4226
+- Loss: 1.5341
 
 ## Model description
 
@@ -140,9 +139,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss |
 |:-------------:|:------:|:----:|:---------------:|
-| 6.2508        | 0.0053 | 3    | 2.4551          |
-| 9.0753        | 0.0105 | 6    | 2.4525          |
-| 14.6351       | 0.0158 | 9    | 2.4226          |
+| 6.3369        | 0.0013 | 3    | 1.5670          |
+| 6.3116        | 0.0026 | 6    | 1.5588          |
+| 6.7369        | 0.0038 | 9    | 1.5341          |
 
 
 ### Framework versions