sohamchougule commited on
Commit
8008d0b
1 Parent(s): 511941f

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -13
README.md CHANGED
@@ -20,7 +20,7 @@ model-index:
20
  metrics:
21
  - name: Rouge1
22
  type: rouge
23
- value: 0.0
24
  ---
25
 
26
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,12 +30,12 @@ should probably proofread and complete it, then remove this comment. -->
30
 
31
  This model is a fine-tuned version of [google/pegasus-large](https://huggingface.co/google/pegasus-large) on the aeslc dataset.
32
  It achieves the following results on the evaluation set:
33
- - Loss: 8.0891
34
- - Rouge1: 0.0
35
- - Rouge2: 0.0
36
- - Rougel: 0.0
37
- - Rougelsum: 0.0
38
- - Gen Len: 1.0
39
 
40
  ## Model description
41
 
@@ -54,9 +54,9 @@ More information needed
54
  ### Training hyperparameters
55
 
56
  The following hyperparameters were used during training:
57
- - learning_rate: 0.0052
58
- - train_batch_size: 4
59
- - eval_batch_size: 4
60
  - seed: 42
61
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
  - lr_scheduler_type: linear
@@ -65,9 +65,9 @@ The following hyperparameters were used during training:
65
 
66
  ### Training results
67
 
68
- | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
69
- |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
70
- | 0.0 | 1.0 | 3609 | 8.0891 | 0.0 | 0.0 | 0.0 | 0.0 | 1.0 |
71
 
72
 
73
  ### Framework versions
 
20
  metrics:
21
  - name: Rouge1
22
  type: rouge
23
+ value: 32.9341
24
  ---
25
 
26
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
30
 
31
  This model is a fine-tuned version of [google/pegasus-large](https://huggingface.co/google/pegasus-large) on the aeslc dataset.
32
  It achieves the following results on the evaluation set:
33
+ - Loss: 3.0274
34
+ - Rouge1: 32.9341
35
+ - Rouge2: 17.7876
36
+ - Rougel: 32.0858
37
+ - Rougelsum: 32.0886
38
+ - Gen Len: 11.6577
39
 
40
  ## Model description
41
 
 
54
  ### Training hyperparameters
55
 
56
  The following hyperparameters were used during training:
57
+ - learning_rate: 2e-05
58
+ - train_batch_size: 2
59
+ - eval_batch_size: 2
60
  - seed: 42
61
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
  - lr_scheduler_type: linear
 
65
 
66
  ### Training results
67
 
68
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
69
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
70
+ | 3.3247 | 1.0 | 7218 | 3.0274 | 32.9341 | 17.7876 | 32.0858 | 32.0886 | 11.6577 |
71
 
72
 
73
  ### Framework versions