bart-large-cnn-finetuned-booksum-chapter

timothyshi/bart-large-cnn-finetuned-booksum-chapter

bart-large-cnn-finetuned-booksum-chapter is a machine learning model from timothyshi. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.

should probably proofread and complete it, then remove this comment. --

Downloads · 30 days

5% of all-time downloads

All-time downloads

1.4K

Public

Repo size

81.3 GB

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.bin1.6 GB · 100%

At a glance

Library: transformers
License: mit
Model type: bart
Access: Public
Created: Mar 4, 2022
Updated: Mar 7, 2022
SHA: 24ef6267

Model card

From the Hugging Face model README

bart-large-cnn-finetuned-booksum-chapter

This model is a fine-tuned version of facebook/bart-large-cnn on the None dataset. It achieves the following results on the evaluation set:

Loss: 3.1373
Rouge1: 18.1222
Rouge2: 3.5783
Rougel: 13.4084
Rougelsum: 13.5832
Gen Len: 63.5121

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

learning_rate: 2e-05
train_batch_size: 1
eval_batch_size: 1
seed: 42
optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
lr_scheduler_type: linear
num_epochs: 1
mixed_precision_training: Native AMP

Training results

Training Loss	Epoch	Step	Validation Loss	Rouge1	Rouge2	Rougel	Rougelsum	Gen Len
2.5297	1.0	23094	3.1373	18.1222	3.5783	13.4084	13.5832	63.5121

Framework versions

Transformers 4.17.0
Pytorch 1.10.0+cu111
Datasets 1.18.3
Tokenizers 0.11.6

Library: transformers
Type: bart
License: mit
Created: Mar 4, 2022
Updated: Mar 7, 2022