Downloads · 30 days
27
7% of all-time downloads
Pegasus_GovReport_model
usakha/Pegasus_GovReport_model
Pegasus_GovReport_model is a summarization model from usakha. Use it when you need a shorter version of a longer text. It is set up for transformers.
learningrate=2e-5 perdevicetrainbatchsize=14 perdeviceevalbatchsize=14 weightdecay=0.01 savetotallimit=3 numtrainepochs=3 predictwithgenerate=True fp16=True
All-time downloads
410
Public
Parameters
571M
11.4 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin2.3 GB · 50%
At a glance
- Task
- Summarization
- Library
- transformers
- Model type
- pegasus
- Access
- Public
- Created
- Jun 22, 2023
- Updated
- Jul 18, 2023
- SHA
- 38db0dc4
Try a prompt
Datasets
Model card
From the Hugging Face model README
Hyperparameters
learning_rate=2e-5
per_device_train_batch_size=14
per_device_eval_batch_size=14
weight_decay=0.01
save_total_limit=3
num_train_epochs=3
predict_with_generate=True
fp16=True
Training Output
global_step=3003,
training_loss=2.0113779983241042,
metrics={'train_runtime': 12268.4376,
'train_samples_per_second': 3.427,
'train_steps_per_second': 0.245,
'total_flos': 1.2147019450889011e+17,
'train_loss': 2.0113779983241042,
'epoch': 3.0}
Training Results
| Epoch | Training Loss | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Bleu | Gen Len |
|---|---|---|---|---|---|---|---|---|
| 1 | 2.035800 | 1.906599 | 0.365400 | 0.150500 | 0.243200 | 0.243500 | 0.366300 | 227.230300 |
| 2 | 1.976100 | 1.878923 | 0.393700 | 0.167800 | 0.263500 | 0.263800 | 0.423600 | 193.114200 |
| 3 | 1.956800 | 1.871454 | 0.409300 | 0.175100 | 0.273400 | 0.273600 | 0.457000 | 172.294500 |
- Task
- Summarization
- Library
- transformers
- Type
- pegasus
- Created
- Jun 22, 2023
- Updated
- Jul 18, 2023