zxc12138 commited on
Commit
205cfd1
1 Parent(s): c2f6560

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -9
README.md CHANGED
@@ -1,8 +1,6 @@
1
  ---
2
  tags:
3
  - generated_from_trainer
4
- datasets:
5
- - samsum
6
  model-index:
7
  - name: pegasus-samsum
8
  results: []
@@ -13,9 +11,9 @@ should probably proofread and complete it, then remove this comment. -->
13
 
14
  # pegasus-samsum
15
 
16
- This model is a fine-tuned version of [google/pegasus-cnn_dailymail](https://huggingface.co/google/pegasus-cnn_dailymail) on the samsum dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 1.4812
19
 
20
  ## Model description
21
 
@@ -49,12 +47,11 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss |
51
  |:-------------:|:-----:|:----:|:---------------:|
52
- | 1.6928 | 0.54 | 500 | 1.4812 |
53
 
54
 
55
  ### Framework versions
56
 
57
- - Transformers 4.26.1
58
- - Pytorch 1.13.1+cu116
59
- - Datasets 2.10.1
60
- - Tokenizers 0.13.2
 
1
  ---
2
  tags:
3
  - generated_from_trainer
 
 
4
  model-index:
5
  - name: pegasus-samsum
6
  results: []
 
11
 
12
  # pegasus-samsum
13
 
14
+ This model is a fine-tuned version of [google/pegasus-cnn_dailymail](https://huggingface.co/google/pegasus-cnn_dailymail) on an unknown dataset.
15
  It achieves the following results on the evaluation set:
16
+ - Loss: 1.4848
17
 
18
  ## Model description
19
 
 
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
+ | 1.6909 | 0.54 | 500 | 1.4848 |
51
 
52
 
53
  ### Framework versions
54
 
55
+ - Transformers 4.28.1
56
+ - Pytorch 2.0.0+cu118
57
+ - Tokenizers 0.13.3