End of training

README.md

@@ -36,7 +36,7 @@ datasets:
     system_prompt: ''
 device_map: auto
 eval_sample_packing: false
-eval_steps: 40
+eval_steps: 50
 flash_attention: true
 gradient_checkpointing: true
 group_by_length: true
@@ -54,7 +54,7 @@ output_dir: /root/.sn56/axolotl/tmp/tiny-random-LlamaForCausalLM
 pad_to_sequence_len: true
 resize_token_embeddings_to_32x: false
 sample_packing: true
-save_steps: 20
+save_steps: 50
 save_total_limit: 2
 sequence_len: 2048
 tokenizer_type: LlamaTokenizerFast
@@ -77,7 +77,7 @@ warmup_ratio: 0.05
 
 This model is a fine-tuned version of [trl-internal-testing/tiny-random-LlamaForCausalLM](https://huggingface.co/trl-internal-testing/tiny-random-LlamaForCausalLM) on the argilla/databricks-dolly-15k-curated-en dataset.
 It achieves the following results on the evaluation set:
-- Loss: 9.1944
+- Loss: 9.1943
 
 ## Model description
 
@@ -114,21 +114,18 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch   | Step | Validation Loss |
 |:-------------:|:-------:|:----:|:---------------:|
 | No log        | 0.0769  | 1    | 10.3764         |
-| 10.3522       | 3.0769  | 40   | 10.3366         |
-| 10.1177       | 6.1538  | 80   | 10.0885         |
-| 9.8887        | 9.2308  | 120  | 9.8677          |
-| 9.688         | 12.3077 | 160  | 9.6724          |
-| 9.5151        | 15.3846 | 200  | 9.5050          |
-| 9.3725        | 18.4615 | 240  | 9.3687          |
-| 9.2678        | 21.5385 | 280  | 9.2734          |
-| 9.2101        | 24.6154 | 320  | 9.2205          |
-| 9.186         | 27.6923 | 360  | 9.2018          |
-| 9.18          | 30.7692 | 400  | 9.1964          |
-| 9.1787        | 33.8462 | 440  | 9.1945          |
-| 9.1768        | 36.9231 | 480  | 9.1941          |
-| 9.1775        | 40.0    | 520  | 9.1938          |
-| 9.1784        | 43.0769 | 560  | 9.1949          |
-| 9.1762        | 46.1538 | 600  | 9.1944          |
+| 10.3159       | 3.8462  | 50   | 10.2852         |
+| 9.998         | 7.6923  | 100  | 9.9738          |
+| 9.7359        | 11.5385 | 150  | 9.7190          |
+| 9.5151        | 15.3846 | 200  | 9.5042          |
+| 9.3407        | 19.2308 | 250  | 9.3411          |
+| 9.2338        | 23.0769 | 300  | 9.2415          |
+| 9.1896        | 26.9231 | 350  | 9.2039          |
+| 9.18          | 30.7692 | 400  | 9.1960          |
+| 9.1777        | 34.6154 | 450  | 9.1957          |
+| 9.1781        | 38.4615 | 500  | 9.1931          |
+| 9.1761        | 42.3077 | 550  | 9.1936          |
+| 9.1762        | 46.1538 | 600  | 9.1943          |
 
 
 ### Framework versions