theojolliffe commited on
Commit
7b1fe41
·
1 Parent(s): 4dc83b0

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +67 -0
README.md ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - rouge
7
+ model-index:
8
+ - name: t5-base-tag-generation-recipes
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # t5-base-tag-generation-recipes
16
+
17
+ This model is a fine-tuned version of [fabiochiu/t5-base-tag-generation](https://huggingface.co/fabiochiu/t5-base-tag-generation) on the None dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 0.6106
20
+ - Rouge1: 76.7317
21
+ - Rouge2: 59.2273
22
+ - Rougel: 74.3318
23
+ - Rougelsum: 74.2181
24
+ - Gen Len: 15.93
25
+
26
+ ## Model description
27
+
28
+ More information needed
29
+
30
+ ## Intended uses & limitations
31
+
32
+ More information needed
33
+
34
+ ## Training and evaluation data
35
+
36
+ More information needed
37
+
38
+ ## Training procedure
39
+
40
+ ### Training hyperparameters
41
+
42
+ The following hyperparameters were used during training:
43
+ - learning_rate: 2e-05
44
+ - train_batch_size: 2
45
+ - eval_batch_size: 2
46
+ - seed: 42
47
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
+ - lr_scheduler_type: linear
49
+ - num_epochs: 4
50
+ - mixed_precision_training: Native AMP
51
+
52
+ ### Training results
53
+
54
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
55
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
56
+ | 1.2596 | 1.0 | 950 | 0.7129 | 75.4613 | 57.2255 | 70.7783 | 70.8028 | 15.56 |
57
+ | 0.7531 | 2.0 | 1900 | 0.6407 | 75.7627 | 58.0085 | 72.5222 | 72.4208 | 15.71 |
58
+ | 0.673 | 3.0 | 2850 | 0.6158 | 76.7922 | 59.4197 | 73.5378 | 73.4624 | 15.71 |
59
+ | 0.6064 | 4.0 | 3800 | 0.6106 | 76.7317 | 59.2273 | 74.3318 | 74.2181 | 15.93 |
60
+
61
+
62
+ ### Framework versions
63
+
64
+ - Transformers 4.26.1
65
+ - Pytorch 1.13.1+cu116
66
+ - Datasets 2.10.1
67
+ - Tokenizers 0.13.2