Mandalor09/PickupLinesGenerator

README.md

@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [EleutherAI/gpt-neo-125m](https://huggingface.co/EleutherAI/gpt-neo-125m) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 3.3814
+- Loss: 1.5085
 
 ## Model description
 
@@ -41,15 +41,16 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 3
+- num_epochs: 4
 
 ### Training results
 
 | Training Loss | Epoch | Step  | Validation Loss |
 |:-------------:|:-----:|:-----:|:---------------:|
-| 3.3518        | 1.0   | 6586  | 3.4081          |
-| 2.6994        | 2.0   | 13172 | 3.3111          |
-| 2.2401        | 3.0   | 19758 | 3.3814          |
+| 2.7252        | 1.0   | 8233  | 2.3500          |
+| 2.3376        | 2.0   | 16466 | 1.9826          |
+| 2.0286        | 3.0   | 24699 | 1.6798          |
+| 1.6443        | 4.0   | 32932 | 1.5085          |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ca733af1435f2f057ae396f6159b311028eccdc493cdf8dad224ef2902cc357d
+oid sha256:f71109ffcb175e65ee44ab2ed1f1b6f5bbe8bbad153e20fb45e0f2312fe899c3
 size 500811336

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8909e4d7768bc59554110a30dc5a14e34472148f6d9da9deb12204f01f23cd18
+oid sha256:d0154eb8d762a337837a4dd7e9a38c25d79fd72f430e98221bd6b31802d026e8
 size 5304