tejaskamtam
commited on
Commit
•
59081f4
1
Parent(s):
6748efb
Model save
Browse files- README.md +112 -0
- generation_config.json +13 -0
README.md
ADDED
@@ -0,0 +1,112 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
base_model: facebook/bart-base
|
4 |
+
tags:
|
5 |
+
- generated_from_trainer
|
6 |
+
metrics:
|
7 |
+
- accuracy
|
8 |
+
model-index:
|
9 |
+
- name: bart-base-finetuned-xe_ey_fae
|
10 |
+
results: []
|
11 |
+
---
|
12 |
+
|
13 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
14 |
+
should probably proofread and complete it, then remove this comment. -->
|
15 |
+
|
16 |
+
# bart-base-finetuned-xe_ey_fae
|
17 |
+
|
18 |
+
This model is a fine-tuned version of [facebook/bart-base](https://huggingface.co/facebook/bart-base) on an unknown dataset.
|
19 |
+
It achieves the following results on the evaluation set:
|
20 |
+
- Loss: 1.3910
|
21 |
+
- Accuracy: 0.7182
|
22 |
+
|
23 |
+
## Model description
|
24 |
+
|
25 |
+
More information needed
|
26 |
+
|
27 |
+
## Intended uses & limitations
|
28 |
+
|
29 |
+
More information needed
|
30 |
+
|
31 |
+
## Training and evaluation data
|
32 |
+
|
33 |
+
More information needed
|
34 |
+
|
35 |
+
## Training procedure
|
36 |
+
|
37 |
+
### Training hyperparameters
|
38 |
+
|
39 |
+
The following hyperparameters were used during training:
|
40 |
+
- learning_rate: 1e-05
|
41 |
+
- train_batch_size: 8
|
42 |
+
- eval_batch_size: 8
|
43 |
+
- seed: 100
|
44 |
+
- gradient_accumulation_steps: 2
|
45 |
+
- total_train_batch_size: 16
|
46 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
47 |
+
- lr_scheduler_type: linear
|
48 |
+
- num_epochs: 3.0
|
49 |
+
- mixed_precision_training: Native AMP
|
50 |
+
|
51 |
+
### Training results
|
52 |
+
|
53 |
+
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
54 |
+
|:-------------:|:-----:|:-----:|:---------------:|:--------:|
|
55 |
+
| 5.4226 | 0.06 | 500 | 3.8138 | 0.3628 |
|
56 |
+
| 4.0408 | 0.12 | 1000 | 3.0576 | 0.4630 |
|
57 |
+
| 3.4979 | 0.18 | 1500 | 2.7016 | 0.5133 |
|
58 |
+
| 3.1691 | 0.24 | 2000 | 2.4880 | 0.5431 |
|
59 |
+
| 2.9564 | 0.3 | 2500 | 2.3309 | 0.5644 |
|
60 |
+
| 2.8078 | 0.35 | 3000 | 2.2320 | 0.5792 |
|
61 |
+
| 2.6741 | 0.41 | 3500 | 2.1506 | 0.5924 |
|
62 |
+
| 2.5323 | 0.47 | 4000 | 1.9846 | 0.6176 |
|
63 |
+
| 2.3678 | 0.53 | 4500 | 1.8813 | 0.6375 |
|
64 |
+
| 2.25 | 0.59 | 5000 | 1.8100 | 0.6497 |
|
65 |
+
| 2.1795 | 0.65 | 5500 | 1.7632 | 0.6579 |
|
66 |
+
| 2.1203 | 0.71 | 6000 | 1.7238 | 0.6646 |
|
67 |
+
| 2.0764 | 0.77 | 6500 | 1.6856 | 0.6713 |
|
68 |
+
| 2.026 | 0.83 | 7000 | 1.6569 | 0.6760 |
|
69 |
+
| 1.9942 | 0.89 | 7500 | 1.6309 | 0.6803 |
|
70 |
+
| 1.9665 | 0.95 | 8000 | 1.6122 | 0.6836 |
|
71 |
+
| 1.9395 | 1.0 | 8500 | 1.5913 | 0.6866 |
|
72 |
+
| 1.9155 | 1.06 | 9000 | 1.5758 | 0.6895 |
|
73 |
+
| 1.8828 | 1.12 | 9500 | 1.5607 | 0.6918 |
|
74 |
+
| 1.8721 | 1.18 | 10000 | 1.5422 | 0.6948 |
|
75 |
+
| 1.8474 | 1.24 | 10500 | 1.5320 | 0.6964 |
|
76 |
+
| 1.8293 | 1.3 | 11000 | 1.5214 | 0.6978 |
|
77 |
+
| 1.8129 | 1.36 | 11500 | 1.5102 | 0.6998 |
|
78 |
+
| 1.8148 | 1.42 | 12000 | 1.5010 | 0.7013 |
|
79 |
+
| 1.7903 | 1.48 | 12500 | 1.4844 | 0.7038 |
|
80 |
+
| 1.7815 | 1.54 | 13000 | 1.4823 | 0.7039 |
|
81 |
+
| 1.7637 | 1.6 | 13500 | 1.4746 | 0.7052 |
|
82 |
+
| 1.7623 | 1.66 | 14000 | 1.4701 | 0.7061 |
|
83 |
+
| 1.7402 | 1.71 | 14500 | 1.4598 | 0.7076 |
|
84 |
+
| 1.7376 | 1.77 | 15000 | 1.4519 | 0.7090 |
|
85 |
+
| 1.7287 | 1.83 | 15500 | 1.4501 | 0.7101 |
|
86 |
+
| 1.7273 | 1.89 | 16000 | 1.4409 | 0.7107 |
|
87 |
+
| 1.7119 | 1.95 | 16500 | 1.4314 | 0.7125 |
|
88 |
+
| 1.7098 | 2.01 | 17000 | 1.4269 | 0.7129 |
|
89 |
+
| 1.6978 | 2.07 | 17500 | 1.4275 | 0.7132 |
|
90 |
+
| 1.698 | 2.13 | 18000 | 1.4218 | 0.7140 |
|
91 |
+
| 1.6837 | 2.19 | 18500 | 1.4151 | 0.7147 |
|
92 |
+
| 1.6908 | 2.25 | 19000 | 1.4137 | 0.7149 |
|
93 |
+
| 1.6902 | 2.31 | 19500 | 1.4085 | 0.7161 |
|
94 |
+
| 1.6741 | 2.36 | 20000 | 1.4121 | 0.7154 |
|
95 |
+
| 1.6823 | 2.42 | 20500 | 1.4037 | 0.7165 |
|
96 |
+
| 1.6692 | 2.48 | 21000 | 1.4039 | 0.7164 |
|
97 |
+
| 1.6669 | 2.54 | 21500 | 1.4015 | 0.7172 |
|
98 |
+
| 1.6613 | 2.6 | 22000 | 1.3979 | 0.7179 |
|
99 |
+
| 1.664 | 2.66 | 22500 | 1.3960 | 0.7180 |
|
100 |
+
| 1.6615 | 2.72 | 23000 | 1.4012 | 0.7172 |
|
101 |
+
| 1.6627 | 2.78 | 23500 | 1.3974 | 0.7178 |
|
102 |
+
| 1.6489 | 2.84 | 24000 | 1.3948 | 0.7182 |
|
103 |
+
| 1.6429 | 2.9 | 24500 | 1.3921 | 0.7184 |
|
104 |
+
| 1.6477 | 2.96 | 25000 | 1.3910 | 0.7182 |
|
105 |
+
|
106 |
+
|
107 |
+
### Framework versions
|
108 |
+
|
109 |
+
- Transformers 4.36.2
|
110 |
+
- Pytorch 2.2.0+cu121
|
111 |
+
- Datasets 2.17.0
|
112 |
+
- Tokenizers 0.15.2
|
generation_config.json
ADDED
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_from_model_config": true,
|
3 |
+
"bos_token_id": 0,
|
4 |
+
"decoder_start_token_id": 2,
|
5 |
+
"early_stopping": true,
|
6 |
+
"eos_token_id": 2,
|
7 |
+
"forced_bos_token_id": 0,
|
8 |
+
"forced_eos_token_id": 2,
|
9 |
+
"no_repeat_ngram_size": 3,
|
10 |
+
"num_beams": 4,
|
11 |
+
"pad_token_id": 1,
|
12 |
+
"transformers_version": "4.36.2"
|
13 |
+
}
|