dialogue_bart

README.md

@@ -1,4 +1,5 @@
 ---
+library_name: transformers
 license: mit
 base_model: facebook/bart-large-cnn
 tags:
@@ -17,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) on the samsum dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2853
+- Loss: 0.1346
 
 ## Model description
 
@@ -42,22 +43,20 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 5
+- num_epochs: 3
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 148  | 0.2371          |
-| No log        | 2.0   | 296  | 0.2405          |
-| No log        | 3.0   | 444  | 0.2506          |
-| 0.1426        | 4.0   | 592  | 0.2660          |
-| 0.1426        | 5.0   | 740  | 0.2853          |
+| No log        | 1.0   | 148  | 0.1932          |
+| No log        | 2.0   | 296  | 0.1462          |
+| No log        | 3.0   | 444  | 0.1346          |
 
 
 ### Framework versions
 
-- Transformers 4.41.2
-- Pytorch 2.3.0+cu121
-- Datasets 2.20.0
+- Transformers 4.44.2
+- Pytorch 2.4.1+cu121
+- Datasets 3.0.1
 - Tokenizers 0.19.1

config.json

@@ -64,7 +64,7 @@
     }
   },
   "torch_dtype": "float32",
-  "transformers_version": "4.41.2",
+  "transformers_version": "4.44.2",
   "use_cache": true,
   "vocab_size": 50264
 }

generation_config.json

@@ -12,5 +12,5 @@
   "no_repeat_ngram_size": 3,
   "num_beams": 4,
   "pad_token_id": 1,
-  "transformers_version": "4.41.2"
+  "transformers_version": "4.44.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a738b1e19404aebc649fc4e8ccc2b4dffbf9862ca1f4ec0a5ead9f965e68c2a8
+oid sha256:97af50142c456c73c48c2a6548b9ee971d85433af4255b9e31e9fc9c52f9839a
 size 1625422896

runs/Sep26_19-01-50_e676c52968d0/events.out.tfevents.1727377349.e676c52968d0.336.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f1c632ec8cca792458f2be90ef48249d67d4e9028cbad82adc5fb4f3b1d5c7ef
+size 6498

runs/Sep26_19-24-41_e676c52968d0/events.out.tfevents.1727378683.e676c52968d0.336.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a9ccda95f82503dfe8592ceffb0a20a4cbf82c382552dd4e41c31d9350c03335
+size 7079

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5ef223d7996243852a9ab52d83185692e85fa960183b2a13fb854fb9bada358c
-size 5112
+oid sha256:d4954b5eddfd2993a9a17c127fdefd64b00de370fd47ae12333b73b3f6e168e0
+size 5176