Pstman commited on
Commit
63612c4
·
1 Parent(s): 50163a4

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -12
README.md CHANGED
@@ -16,12 +16,12 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 2.3750
20
- - Rouge1: 0.3111
21
- - Rouge2: 0.1149
22
- - Rougel: 0.2256
23
- - Rougelsum: 0.2256
24
- - Gen Len: 65.5641
25
 
26
  ## Model description
27
 
@@ -41,8 +41,8 @@ More information needed
41
 
42
  The following hyperparameters were used during training:
43
  - learning_rate: 2e-05
44
- - train_batch_size: 4
45
- - eval_batch_size: 4
46
  - seed: 42
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
@@ -53,10 +53,10 @@ The following hyperparameters were used during training:
53
 
54
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
55
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
56
- | 1.8443 | 1.0 | 1280 | 1.8708 | 0.3152 | 0.1199 | 0.2279 | 0.2279 | 63.8984 |
57
- | 1.244 | 2.0 | 2560 | 1.9406 | 0.3174 | 0.1199 | 0.2303 | 0.2305 | 63.8938 |
58
- | 0.8394 | 3.0 | 3840 | 2.1484 | 0.3118 | 0.1167 | 0.2253 | 0.2254 | 65.0891 |
59
- | 0.5744 | 4.0 | 5120 | 2.3750 | 0.3111 | 0.1149 | 0.2256 | 0.2256 | 65.5641 |
60
 
61
 
62
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 2.1211
20
+ - Rouge1: 0.3212
21
+ - Rouge2: 0.1225
22
+ - Rougel: 0.2329
23
+ - Rougelsum: 0.2327
24
+ - Gen Len: 65.4998
25
 
26
  ## Model description
27
 
 
41
 
42
  The following hyperparameters were used during training:
43
  - learning_rate: 2e-05
44
+ - train_batch_size: 8
45
+ - eval_batch_size: 8
46
  - seed: 42
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
 
53
 
54
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
55
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
56
+ | 1.8113 | 1.0 | 1134 | 1.7859 | 0.3221 | 0.1245 | 0.2348 | 0.2348 | 64.0653 |
57
+ | 1.3327 | 2.0 | 2268 | 1.8142 | 0.3248 | 0.1255 | 0.2368 | 0.2368 | 64.2325 |
58
+ | 0.9968 | 3.0 | 3402 | 1.9795 | 0.3215 | 0.1236 | 0.2335 | 0.2335 | 65.0935 |
59
+ | 0.7786 | 4.0 | 4536 | 2.1211 | 0.3212 | 0.1225 | 0.2329 | 0.2327 | 65.4998 |
60
 
61
 
62
  ### Framework versions