update model card README.md

README.md

@@ -1,4 +1,5 @@
 ---
+license: apache-2.0
 tags:
 - generated_from_trainer
 model-index:
@@ -11,9 +12,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # HintsGenerator
 
-This model is a fine-tuned version of [patrickvonplaten/t5-tiny-random](https://huggingface.co/patrickvonplaten/t5-tiny-random) on an unknown dataset.
+This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 5.7914
+- Loss: 2.1726
 
 ## Model description
 
@@ -33,32 +34,29 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 16
+- train_batch_size: 128
 - eval_batch_size: 8
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 10
+- num_epochs: 7
 
 ### Training results
 
-| Training Loss | Epoch | Step  | Validation Loss |
-|:-------------:|:-----:|:-----:|:---------------:|
-| 4.8644        | 1.0   | 4212  | 7.3992          |
-| 4.2267        | 2.0   | 8424  | 6.8927          |
-| 3.8213        | 3.0   | 12636 | 6.5816          |
-| 3.5707        | 4.0   | 16848 | 6.3507          |
-| 3.3617        | 5.0   | 21060 | 6.1117          |
-| 3.2614        | 6.0   | 25272 | 5.9754          |
-| 3.1676        | 7.0   | 29484 | 5.8835          |
-| 3.1363        | 8.0   | 33696 | 5.8340          |
-| 3.1046        | 9.0   | 37908 | 5.7960          |
-| 3.1239        | 10.0  | 42120 | 5.7914          |
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 3.3501        | 1.0   | 527  | 2.3548          |
+| 2.6694        | 2.0   | 1054 | 2.2615          |
+| 2.5871        | 3.0   | 1581 | 2.2202          |
+| 2.5442        | 4.0   | 2108 | 2.1957          |
+| 2.5154        | 5.0   | 2635 | 2.1837          |
+| 2.4983        | 6.0   | 3162 | 2.1757          |
+| 2.4929        | 7.0   | 3689 | 2.1726          |
 
 
 ### Framework versions
 
-- Transformers 4.28.1
-- Pytorch 2.0.0+cu118
+- Transformers 4.30.0
+- Pytorch 2.0.1+cu118
 - Datasets 2.12.0
 - Tokenizers 0.13.3