gokulsrinivasagan commited on
Commit
f3a271d
·
verified ·
1 Parent(s): b16cf71

End of training

Browse files
README.md CHANGED
@@ -1,15 +1,33 @@
1
  ---
2
  library_name: transformers
 
 
3
  license: apache-2.0
4
  base_model: gokulsrinivasagan/tinybert_base_train_kd
5
  tags:
6
  - generated_from_trainer
 
 
7
  metrics:
8
  - accuracy
9
  - f1
10
  model-index:
11
  - name: tinybert_base_train_kd_mrpc
12
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  ---
14
 
15
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,12 +35,12 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # tinybert_base_train_kd_mrpc
19
 
20
- This model is a fine-tuned version of [gokulsrinivasagan/tinybert_base_train_kd](https://huggingface.co/gokulsrinivasagan/tinybert_base_train_kd) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.7977
23
- - Accuracy: 0.7549
24
- - F1: 0.8214
25
- - Combined Score: 0.7882
26
 
27
  ## Model description
28
 
 
1
  ---
2
  library_name: transformers
3
+ language:
4
+ - en
5
  license: apache-2.0
6
  base_model: gokulsrinivasagan/tinybert_base_train_kd
7
  tags:
8
  - generated_from_trainer
9
+ datasets:
10
+ - glue
11
  metrics:
12
  - accuracy
13
  - f1
14
  model-index:
15
  - name: tinybert_base_train_kd_mrpc
16
+ results:
17
+ - task:
18
+ name: Text Classification
19
+ type: text-classification
20
+ dataset:
21
+ name: GLUE MRPC
22
+ type: glue
23
+ args: mrpc
24
+ metrics:
25
+ - name: Accuracy
26
+ type: accuracy
27
+ value: 0.7916666666666666
28
+ - name: F1
29
+ type: f1
30
+ value: 0.856175972927242
31
  ---
32
 
33
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
35
 
36
  # tinybert_base_train_kd_mrpc
37
 
38
+ This model is a fine-tuned version of [gokulsrinivasagan/tinybert_base_train_kd](https://huggingface.co/gokulsrinivasagan/tinybert_base_train_kd) on the GLUE MRPC dataset.
39
  It achieves the following results on the evaluation set:
40
+ - Loss: 0.5067
41
+ - Accuracy: 0.7917
42
+ - F1: 0.8562
43
+ - Combined Score: 0.8239
44
 
45
  ## Model description
46
 
all_results.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.0,
3
+ "eval_accuracy": 0.7916666666666666,
4
+ "eval_combined_score": 0.8239213197969544,
5
+ "eval_f1": 0.856175972927242,
6
+ "eval_loss": 0.5067150592803955,
7
+ "eval_runtime": 0.2507,
8
+ "eval_samples": 408,
9
+ "eval_samples_per_second": 1627.171,
10
+ "eval_steps_per_second": 7.976,
11
+ "total_flos": 2186506882215936.0,
12
+ "train_loss": 0.3955998191127071,
13
+ "train_runtime": 48.5344,
14
+ "train_samples": 3668,
15
+ "train_samples_per_second": 3778.761,
16
+ "train_steps_per_second": 15.453
17
+ }
eval_results.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.0,
3
+ "eval_accuracy": 0.7916666666666666,
4
+ "eval_combined_score": 0.8239213197969544,
5
+ "eval_f1": 0.856175972927242,
6
+ "eval_loss": 0.5067150592803955,
7
+ "eval_runtime": 0.2507,
8
+ "eval_samples": 408,
9
+ "eval_samples_per_second": 1627.171,
10
+ "eval_steps_per_second": 7.976
11
+ }
logs/events.out.tfevents.1744834148.ki-g0008.514628.21 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f6497d904e3e37a6408e5619f17dd826870dbb7a19ff6d14acfdf277b691b6db
3
+ size 515
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.0,
3
+ "total_flos": 2186506882215936.0,
4
+ "train_loss": 0.3955998191127071,
5
+ "train_runtime": 48.5344,
6
+ "train_samples": 3668,
7
+ "train_samples_per_second": 3778.761,
8
+ "train_steps_per_second": 15.453
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,214 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 60,
3
+ "best_metric": 0.5067150592803955,
4
+ "best_model_checkpoint": "tinybert_base_train_kd_mrpc/checkpoint-60",
5
+ "epoch": 9.0,
6
+ "eval_steps": 500,
7
+ "global_step": 135,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 1.0,
14
+ "grad_norm": 1.0588346719741821,
15
+ "learning_rate": 4.906666666666667e-05,
16
+ "loss": 0.6588,
17
+ "step": 15
18
+ },
19
+ {
20
+ "epoch": 1.0,
21
+ "eval_accuracy": 0.6666666666666666,
22
+ "eval_combined_score": 0.7160919540229884,
23
+ "eval_f1": 0.7655172413793103,
24
+ "eval_loss": 0.6283461451530457,
25
+ "eval_runtime": 0.2187,
26
+ "eval_samples_per_second": 1865.558,
27
+ "eval_steps_per_second": 9.145,
28
+ "step": 15
29
+ },
30
+ {
31
+ "epoch": 2.0,
32
+ "grad_norm": 1.6302603483200073,
33
+ "learning_rate": 4.806666666666667e-05,
34
+ "loss": 0.5998,
35
+ "step": 30
36
+ },
37
+ {
38
+ "epoch": 2.0,
39
+ "eval_accuracy": 0.696078431372549,
40
+ "eval_combined_score": 0.7508605323006945,
41
+ "eval_f1": 0.8056426332288401,
42
+ "eval_loss": 0.5731440782546997,
43
+ "eval_runtime": 0.2338,
44
+ "eval_samples_per_second": 1744.854,
45
+ "eval_steps_per_second": 8.553,
46
+ "step": 30
47
+ },
48
+ {
49
+ "epoch": 3.0,
50
+ "grad_norm": 5.0050811767578125,
51
+ "learning_rate": 4.706666666666667e-05,
52
+ "loss": 0.5433,
53
+ "step": 45
54
+ },
55
+ {
56
+ "epoch": 3.0,
57
+ "eval_accuracy": 0.7401960784313726,
58
+ "eval_combined_score": 0.786764705882353,
59
+ "eval_f1": 0.8333333333333334,
60
+ "eval_loss": 0.5613774657249451,
61
+ "eval_runtime": 0.2224,
62
+ "eval_samples_per_second": 1834.58,
63
+ "eval_steps_per_second": 8.993,
64
+ "step": 45
65
+ },
66
+ {
67
+ "epoch": 4.0,
68
+ "grad_norm": 10.076091766357422,
69
+ "learning_rate": 4.606666666666667e-05,
70
+ "loss": 0.4574,
71
+ "step": 60
72
+ },
73
+ {
74
+ "epoch": 4.0,
75
+ "eval_accuracy": 0.7916666666666666,
76
+ "eval_combined_score": 0.8239213197969544,
77
+ "eval_f1": 0.856175972927242,
78
+ "eval_loss": 0.5067150592803955,
79
+ "eval_runtime": 0.2267,
80
+ "eval_samples_per_second": 1799.599,
81
+ "eval_steps_per_second": 8.822,
82
+ "step": 60
83
+ },
84
+ {
85
+ "epoch": 5.0,
86
+ "grad_norm": 12.540026664733887,
87
+ "learning_rate": 4.5066666666666667e-05,
88
+ "loss": 0.3525,
89
+ "step": 75
90
+ },
91
+ {
92
+ "epoch": 5.0,
93
+ "eval_accuracy": 0.7573529411764706,
94
+ "eval_combined_score": 0.7888398099711753,
95
+ "eval_f1": 0.8203266787658802,
96
+ "eval_loss": 0.5203803777694702,
97
+ "eval_runtime": 0.2188,
98
+ "eval_samples_per_second": 1864.761,
99
+ "eval_steps_per_second": 9.141,
100
+ "step": 75
101
+ },
102
+ {
103
+ "epoch": 6.0,
104
+ "grad_norm": 10.685439109802246,
105
+ "learning_rate": 4.406666666666667e-05,
106
+ "loss": 0.3015,
107
+ "step": 90
108
+ },
109
+ {
110
+ "epoch": 6.0,
111
+ "eval_accuracy": 0.7426470588235294,
112
+ "eval_combined_score": 0.7904298468231669,
113
+ "eval_f1": 0.8382126348228043,
114
+ "eval_loss": 0.829555332660675,
115
+ "eval_runtime": 0.2251,
116
+ "eval_samples_per_second": 1812.326,
117
+ "eval_steps_per_second": 8.884,
118
+ "step": 90
119
+ },
120
+ {
121
+ "epoch": 7.0,
122
+ "grad_norm": 5.721017360687256,
123
+ "learning_rate": 4.3066666666666665e-05,
124
+ "loss": 0.2978,
125
+ "step": 105
126
+ },
127
+ {
128
+ "epoch": 7.0,
129
+ "eval_accuracy": 0.75,
130
+ "eval_combined_score": 0.7950626959247649,
131
+ "eval_f1": 0.8401253918495298,
132
+ "eval_loss": 0.7863379120826721,
133
+ "eval_runtime": 0.2298,
134
+ "eval_samples_per_second": 1775.245,
135
+ "eval_steps_per_second": 8.702,
136
+ "step": 105
137
+ },
138
+ {
139
+ "epoch": 8.0,
140
+ "grad_norm": 11.30908203125,
141
+ "learning_rate": 4.206666666666667e-05,
142
+ "loss": 0.2163,
143
+ "step": 120
144
+ },
145
+ {
146
+ "epoch": 8.0,
147
+ "eval_accuracy": 0.7573529411764706,
148
+ "eval_combined_score": 0.7940610859728507,
149
+ "eval_f1": 0.8307692307692308,
150
+ "eval_loss": 0.7096235752105713,
151
+ "eval_runtime": 0.2225,
152
+ "eval_samples_per_second": 1833.884,
153
+ "eval_steps_per_second": 8.99,
154
+ "step": 120
155
+ },
156
+ {
157
+ "epoch": 9.0,
158
+ "grad_norm": 4.002461910247803,
159
+ "learning_rate": 4.106666666666667e-05,
160
+ "loss": 0.1329,
161
+ "step": 135
162
+ },
163
+ {
164
+ "epoch": 9.0,
165
+ "eval_accuracy": 0.7549019607843137,
166
+ "eval_combined_score": 0.7881652661064426,
167
+ "eval_f1": 0.8214285714285714,
168
+ "eval_loss": 0.7977427840232849,
169
+ "eval_runtime": 0.2192,
170
+ "eval_samples_per_second": 1861.282,
171
+ "eval_steps_per_second": 9.124,
172
+ "step": 135
173
+ },
174
+ {
175
+ "epoch": 9.0,
176
+ "step": 135,
177
+ "total_flos": 2186506882215936.0,
178
+ "train_loss": 0.3955998191127071,
179
+ "train_runtime": 48.5344,
180
+ "train_samples_per_second": 3778.761,
181
+ "train_steps_per_second": 15.453
182
+ }
183
+ ],
184
+ "logging_steps": 1,
185
+ "max_steps": 750,
186
+ "num_input_tokens_seen": 0,
187
+ "num_train_epochs": 50,
188
+ "save_steps": 500,
189
+ "stateful_callbacks": {
190
+ "EarlyStoppingCallback": {
191
+ "args": {
192
+ "early_stopping_patience": 5,
193
+ "early_stopping_threshold": 0.0
194
+ },
195
+ "attributes": {
196
+ "early_stopping_patience_counter": 5
197
+ }
198
+ },
199
+ "TrainerControl": {
200
+ "args": {
201
+ "should_epoch_stop": false,
202
+ "should_evaluate": false,
203
+ "should_log": false,
204
+ "should_save": true,
205
+ "should_training_stop": true
206
+ },
207
+ "attributes": {}
208
+ }
209
+ },
210
+ "total_flos": 2186506882215936.0,
211
+ "train_batch_size": 256,
212
+ "trial_name": null,
213
+ "trial_params": null
214
+ }