End of training

README.md

@@ -1,8 +1,8 @@
 ---
 license: mit
+base_model: jinunyachhyon/MobileLLM_Finetune_onDialogueDataset
 tags:
 - generated_from_trainer
-base_model: currentlyexhausted/mobile-llm
 model-index:
 - name: MobileLLM_Finetune_onDialogueDataset
   results: []
@@ -13,7 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # MobileLLM_Finetune_onDialogueDataset
 
-This model is a fine-tuned version of [currentlyexhausted/mobile-llm](https://huggingface.co/currentlyexhausted/mobile-llm) on an unknown dataset.
+This model is a fine-tuned version of [jinunyachhyon/MobileLLM_Finetune_onDialogueDataset](https://huggingface.co/jinunyachhyon/MobileLLM_Finetune_onDialogueDataset) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0765
 
 ## Model description
 
@@ -38,11 +40,23 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 2
+- num_epochs: 5
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch | Step  | Validation Loss |
+|:-------------:|:-----:|:-----:|:---------------:|
+| 0.0891        | 1.0   | 7355  | 0.0780          |
+| 0.0855        | 2.0   | 14710 | 0.0771          |
+| 0.0799        | 3.0   | 22065 | 0.0767          |
+| 0.0804        | 4.0   | 29420 | 0.0764          |
+| 0.0777        | 5.0   | 36775 | 0.0765          |
+
+
 ### Framework versions
 
-- Transformers 4.38.2
-- Pytorch 2.2.1+cu121
+- Transformers 4.39.3
+- Pytorch 2.1.2
+- Datasets 2.18.0
 - Tokenizers 0.15.2

generation_config.json

@@ -3,5 +3,5 @@
   "decoder_start_token_id": 0,
   "eos_token_id": 1,
   "pad_token_id": 0,
-  "transformers_version": "4.38.2"
+  "transformers_version": "4.39.3"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0c77b57a4f1fc12af6ff757d1dfbfa8b83864c7b7c302e6298d1de601b53fe9c
+oid sha256:0655e50340054bcda18df2afc1cda5f5651318f4167a8f1d5d278aa507ec80f2
 size 990345064