elliotthwang commited on
Commit
2017b16
1 Parent(s): 741043c

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -11
README.md CHANGED
@@ -19,7 +19,7 @@ model-index:
19
  metrics:
20
  - name: Rouge1
21
  type: rouge
22
- value: 0.7966
23
  ---
24
 
25
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -29,12 +29,12 @@ should probably proofread and complete it, then remove this comment. -->
29
 
30
  This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the xlsum dataset.
31
  It achieves the following results on the evaluation set:
32
- - Loss: 1.2848
33
- - Rouge1: 0.7966
34
- - Rouge2: 0.0942
35
- - Rougel: 0.7989
36
- - Rougelsum: 0.7967
37
- - Gen Len: 7.0854
38
 
39
  ## Model description
40
 
@@ -59,14 +59,19 @@ The following hyperparameters were used during training:
59
  - seed: 42
60
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
61
  - lr_scheduler_type: linear
62
- - num_epochs: 1
63
  - mixed_precision_training: Native AMP
64
 
65
  ### Training results
66
 
67
- | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
68
- |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
69
- | 1.4457 | 1.0 | 2336 | 1.2848 | 0.7966 | 0.0942 | 0.7989 | 0.7967 | 7.0854 |
 
 
 
 
 
70
 
71
 
72
  ### Framework versions
 
19
  metrics:
20
  - name: Rouge1
21
  type: rouge
22
+ value: 0.8887
23
  ---
24
 
25
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
29
 
30
  This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the xlsum dataset.
31
  It achieves the following results on the evaluation set:
32
+ - Loss: 1.2061
33
+ - Rouge1: 0.8887
34
+ - Rouge2: 0.0671
35
+ - Rougel: 0.889
36
+ - Rougelsum: 0.8838
37
+ - Gen Len: 6.8779
38
 
39
  ## Model description
40
 
 
59
  - seed: 42
60
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
61
  - lr_scheduler_type: linear
62
+ - num_epochs: 6
63
  - mixed_precision_training: Native AMP
64
 
65
  ### Training results
66
 
67
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
68
+ |:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
69
+ | 1.4231 | 1.0 | 2336 | 1.2586 | 0.711 | 0.0528 | 0.7029 | 0.7053 | 7.3368 |
70
+ | 1.378 | 2.0 | 4672 | 1.2281 | 0.9688 | 0.05 | 0.9574 | 0.9656 | 7.0392 |
71
+ | 1.3567 | 3.0 | 7008 | 1.2182 | 0.9534 | 0.1035 | 0.9531 | 0.9472 | 6.7437 |
72
+ | 1.3339 | 4.0 | 9344 | 1.2096 | 0.9969 | 0.0814 | 0.9969 | 0.9938 | 7.4503 |
73
+ | 1.3537 | 5.0 | 11680 | 1.2072 | 0.8429 | 0.0742 | 0.8372 | 0.838 | 6.8049 |
74
+ | 1.3351 | 6.0 | 14016 | 1.2061 | 0.8887 | 0.0671 | 0.889 | 0.8838 | 6.8779 |
75
 
76
 
77
  ### Framework versions