End of training

README.md

@@ -4,18 +4,18 @@ base_model: facebook/bart-base
 tags:
 - generated_from_trainer
 model-index:
-- name: Bart_complaint
+- name: Bart_Complaint
   results: []
 ---
 
 <!-- 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_complaint
+# Bart_Complaint
 
 This model is a fine-tuned version of [facebook/bart-base](https://huggingface.co/facebook/bart-base) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0104
+- Loss: 0.1493
 
 ## Model description
 
@@ -35,21 +35,27 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 10
-- eval_batch_size: 10
+- train_batch_size: 24
+- eval_batch_size: 24
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 500
-- num_epochs: 5
+- num_epochs: 3
+- mixed_precision_training: Native AMP
 
 ### Training results
 
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 0.3009        | 1.0   | 212  | 0.2204          |
+| 0.2167        | 2.0   | 424  | 0.1964          |
+| 0.1449        | 3.0   | 636  | 0.1493          |
 
 
 ### Framework versions
 
-- Transformers 4.41.0
-- Pytorch 2.3.0+cu121
+- Transformers 4.41.1
+- Pytorch 1.13.1+cu117
 - Datasets 2.19.1
 - Tokenizers 0.19.1

config.json

@@ -69,7 +69,7 @@
     }
   },
   "torch_dtype": "float32",
-  "transformers_version": "4.41.0",
+  "transformers_version": "4.41.1",
   "use_cache": true,
   "vocab_size": 50265
 }

generation_config.json

@@ -9,5 +9,5 @@
   "no_repeat_ngram_size": 3,
   "num_beams": 4,
   "pad_token_id": 1,
-  "transformers_version": "4.41.0"
+  "transformers_version": "4.41.1"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5adc498a81d9ac20d8327489a5f6536edcbfcf5cfadd0a6849a105768ab43c69
+oid sha256:1dc0e6d1d0efa9abba264ae93b7bc2147b05fe9cad18b8e3ec2b32799b7c9ea6
 size 557912620

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bc08dfc2ae140133064f6217a0404596a2fd6265f17d02bc05045fa7bfe61859
-size 5048
+oid sha256:0a6d3d5466835de2d054923d63e8ab0cf2d664a82b0f1c551aca328156bf3643
+size 4603