tejaskamtam commited on
Commit
59081f4
1 Parent(s): 6748efb

Model save

Browse files
Files changed (2) hide show
  1. README.md +112 -0
  2. 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
+ }