End of training

README.md

@@ -1,6 +1,6 @@
 ---
 license: mit
-base_model: gpt2
+base_model: openai-community/gpt2
 tags:
 - generated_from_trainer
 model-index:
@@ -13,9 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # gpt2-wikitext2
 
-This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
+This model is a fine-tuned version of [openai-community/gpt2](https://huggingface.co/openai-community/gpt2) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 5.0568
+- Loss: 5.9799
 
 ## Model description
 
@@ -46,16 +46,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 65   | 6.2980          |
-| No log        | 2.0   | 130  | 5.7891          |
-| No log        | 3.0   | 195  | 5.4985          |
-| No log        | 4.0   | 260  | 5.3319          |
-| No log        | 5.0   | 325  | 5.2223          |
-| No log        | 6.0   | 390  | 5.1515          |
-| No log        | 7.0   | 455  | 5.1093          |
-| 5.2435        | 8.0   | 520  | 5.0772          |
-| 5.2435        | 9.0   | 585  | 5.0629          |
-| 5.2435        | 10.0  | 650  | 5.0568          |
+| No log        | 1.0   | 55   | 8.6529          |
+| No log        | 2.0   | 110  | 7.5256          |
+| No log        | 3.0   | 165  | 6.9506          |
+| No log        | 4.0   | 220  | 6.5920          |
+| No log        | 5.0   | 275  | 6.3522          |
+| No log        | 6.0   | 330  | 6.1985          |
+| No log        | 7.0   | 385  | 6.0856          |
+| No log        | 8.0   | 440  | 6.0268          |
+| No log        | 9.0   | 495  | 5.9909          |
+| 6.7444        | 10.0  | 550  | 5.9799          |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:38ba402a0bbffcf14f21b56a6414ef49d6f6436ab4f64b675775827a504cc574
+oid sha256:f09c375e256c62f5d1071285ab8298b8612eb9c56174b27e7be24405132e4f60
 size 497774208