End of training

README.md

@@ -1,10 +1,10 @@
 ---
 license: mit
+base_model: facebook/bart-large-xsum
 tags:
 - generated_from_trainer
-metrics:
-- rouge
-base_model: facebook/bart-large-xsum
+datasets:
+- text-summatization
 model-index:
 - name: ft-facebook-bart-large-xsum-on-samsum
   results: []
@@ -15,14 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # ft-facebook-bart-large-xsum-on-samsum
 
-This model is a fine-tuned version of [facebook/bart-large-xsum](https://huggingface.co/facebook/bart-large-xsum) on the None dataset.
-It achieves the following results on the evaluation set:
-- Loss: 1.4691
-- Rouge1: 51.1221
-- Rouge2: 25.9275
-- Rougel: 41.5903
-- Rougelsum: 46.7354
-- Gen Len: 26.89
+This model is a fine-tuned version of [facebook/bart-large-xsum](https://huggingface.co/facebook/bart-large-xsum) on the Samsum dataset.
 
 ## Model description
 
@@ -54,9 +47,9 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2  | Rougel  | Rougelsum | Gen Len |
-|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
-| No log        | 0.22  | 100  | 1.5053          | 49.3903 | 24.5197 | 40.7578 | 45.2724   | 26.3272 |
+| Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2 | Rougel  | Rougelsum | Gen Len |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|:------:|:-------:|:---------:|:-------:|
+| No log        | 0.22  | 100  | 1.5054          | 49.5352 | 24.54  | 40.7807 | 45.3634   | 26.4444 |
 
 
 ### Framework versions

generation_config.json

@@ -1,9 +1,13 @@
 {
+  "bos_token_id": 0,
+  "decoder_start_token_id": 2,
   "early_stopping": true,
+  "eos_token_id": 2,
   "forced_eos_token_id": 2,
   "max_length": 62,
   "min_length": 11,
   "no_repeat_ngram_size": 3,
   "num_beams": 6,
+  "pad_token_id": 1,
   "transformers_version": "4.37.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:388374e630771e997bf530aece6d276cb7a43934219134d0812ffbdf0cbfaf1b
+oid sha256:72340630ae1cbf4c0eec78646e581be23d9860db212f7a9d21d5b4dbe20d5d6b
 size 1625422896

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5ad9f76233b0c45eee265be4dbb2b5fd8493ad0e864c4bb486e6574c3b617b45
-size 4920
+oid sha256:fa9176f2fc9647455e45407699680ea8aa4ceb425b2e7b8b346469463685c259
+size 4856