jrtec
/

jrtec-gpt2-superheroes-name-generator

Text Generation

Generated from Trainer

text-generation-inference

Model card Files Files and versions Metrics Training metrics Community

jonra1993 commited on Jan 8, 2023

Commit

62d4bc4

·

1 Parent(s): 3118640

update model card README.md

Files changed (1) hide show

README.md +7 -4

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 17.2505
 ## Model description
@@ -39,14 +39,17 @@ 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: 10
 ### Training results
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 2.1502        | 3.45  | 500  | 17.2505         |
-| 2.1151        | 6.9   | 1000 | 18.1086         |
 ### Framework versions

 This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the None dataset.
 It achieves the following results on the evaluation set:
+- Loss: 16.5324
 ## Model description
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
+- num_epochs: 20
 ### Training results
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
+| 3.3728        | 3.45  | 500  | 14.0617         |
+| 2.0865        | 6.9   | 1000 | 14.7614         |
+| 1.9246        | 10.34 | 1500 | 15.6696         |
+| 1.8718        | 13.79 | 2000 | 16.2823         |
+| 1.8267        | 17.24 | 2500 | 16.5324         |
 ### Framework versions