CoolFace
Modelpublic

yoeel/bart-cnn-summarizer

sourceHugging Faceapache-2.0updated 21d agoView on Hugging Face
0likes500downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

bart-cnn-summarizer

This model is a fine-tuned version of facebook/bart-base on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 3.2134

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: 5e-05
  • trainbatchsize: 8
  • evalbatchsize: 8
  • seed: 42
  • gradientaccumulationsteps: 4
  • totaltrainbatch_size: 32
  • optimizer: Use OptimizerNames.ADAMWTORCHFUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lrschedulertype: linear
  • lrschedulerwarmup_steps: 100
  • num_epochs: 1
  • mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation Loss
19.49010.01781003.7653
17.55430.03562003.6700
17.31400.05333003.5888
16.96650.07114003.5865
16.79500.08895003.5545
16.41160.10676003.5126
16.61960.12447003.4994
16.48610.14228003.4613
16.33990.169003.4798
16.13880.177810003.5040
16.08950.195611003.4076
16.03610.213312003.4577
15.98890.231113003.4254
16.09830.248914003.3988
15.99380.266715003.4130
15.73510.284416003.4220
15.95610.302217003.3938
15.92530.3218003.3993
15.88450.337819003.3602
15.66290.355620003.3580
15.43230.373321003.3202
15.76000.391122003.3595
15.62330.408923003.3359
15.51260.426724003.3498
15.43040.444425003.3059
15.46070.462226003.3204
15.48000.4827003.3184
15.34040.497828003.3151
15.31500.515629003.3021
15.49260.533330003.2872
15.37380.551131003.3185
15.52170.568932003.2836
15.31600.586733003.2977
15.39710.604434003.2818
15.24910.622235003.2802
15.26160.6436003.2686
15.35590.657837003.2731
15.16600.675638003.2616
15.14040.693339003.2743
15.11530.711140003.2549
15.22210.728941003.2640
15.00890.746742003.2657
15.23020.764443003.2426
15.09450.782244003.2425
15.09070.845003.2428
15.00470.817846003.2483
15.14610.835647003.2180
15.05220.853348003.2334
14.99800.871149003.2294
15.02230.888950003.2200
14.80980.906751003.2266
14.98310.924452003.2287
14.99490.942253003.2185
14.88420.9654003.2211
14.93450.977855003.2116
15.01630.995656003.2134

Framework versions

  • Transformers 5.0.0
  • Pytorch 2.10.0+cu128
  • Datasets 5.0.0
  • Tokenizers 0.22.2