End of training

README.md

@@ -1,6 +1,6 @@
 ---
 license: mit
-base_model: gpt2
+base_model: ajay141/text-gen
 tags:
 - generated_from_trainer
 model-index:
@@ -13,9 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # text-gen
 
-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 [ajay141/text-gen](https://huggingface.co/ajay141/text-gen) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 3.9695
+- Loss: 3.3603
 
 ## Model description
 
@@ -49,8 +49,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 4.3697        | 1.0   | 333  | 4.3313          |
-| 3.9941        | 2.0   | 666  | 3.9695          |
+| 3.5553        | 1.0   | 333  | 3.5634          |
+| 3.3094        | 2.0   | 666  | 3.3603          |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:524584faf01cf710ecdb3c51d5c1fbf4adeed7a80df34d66bae2f355fccfbf24
+oid sha256:7d487191d14a4deb3c72ece5dcacb2267ea78c0720cdb78b89815884b444918c
 size 503128704

runs/Apr02_05-20-44_223ea75a1d88/events.out.tfevents.1712035277.223ea75a1d88.15838.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6cc5599ee11892fc6d850a786ba69eca583eb5157133b2aca30cd1e8d0b09f26
-size 15593
+oid sha256:943b48b0c4ee0b75085f9a3918a1ba70b437cb0c33d0e0eb666ec72b3eee3019
+size 19594