gokuls commited on
Commit
d4c1dc5
1 Parent(s): 6685360

End of training

Browse files
README.md CHANGED
@@ -1,4 +1,6 @@
1
  ---
 
 
2
  tags:
3
  - generated_from_trainer
4
  datasets:
@@ -12,7 +14,7 @@ model-index:
12
  name: Text Classification
13
  type: text-classification
14
  dataset:
15
- name: glue
16
  type: glue
17
  config: stsb
18
  split: validation
@@ -20,7 +22,7 @@ model-index:
20
  metrics:
21
  - name: Spearmanr
22
  type: spearmanr
23
- value: 0.750517182024731
24
  ---
25
 
26
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -28,12 +30,12 @@ should probably proofread and complete it, then remove this comment. -->
28
 
29
  # hBERTv1_new_pretrain_w_init_48_stsb
30
 
31
- This model is a fine-tuned version of [gokuls/bert_12_layer_model_v1_complete_training_new_wt_init_48](https://huggingface.co/gokuls/bert_12_layer_model_v1_complete_training_new_wt_init_48) on the glue dataset.
32
  It achieves the following results on the evaluation set:
33
- - Loss: 1.0470
34
- - Pearson: 0.7517
35
- - Spearmanr: 0.7505
36
- - Combined Score: 0.7511
37
 
38
  ## Model description
39
 
 
1
  ---
2
+ language:
3
+ - en
4
  tags:
5
  - generated_from_trainer
6
  datasets:
 
14
  name: Text Classification
15
  type: text-classification
16
  dataset:
17
+ name: GLUE STSB
18
  type: glue
19
  config: stsb
20
  split: validation
 
22
  metrics:
23
  - name: Spearmanr
24
  type: spearmanr
25
+ value: 0.7471924680940966
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
30
 
31
  # hBERTv1_new_pretrain_w_init_48_stsb
32
 
33
+ This model is a fine-tuned version of [gokuls/bert_12_layer_model_v1_complete_training_new_wt_init_48](https://huggingface.co/gokuls/bert_12_layer_model_v1_complete_training_new_wt_init_48) on the GLUE STSB dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.9800
36
+ - Pearson: 0.7515
37
+ - Spearmanr: 0.7472
38
+ - Combined Score: 0.7493
39
 
40
  ## Model description
41
 
all_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 14.0,
3
+ "eval_combined_score": 0.7493306108739174,
4
+ "eval_loss": 0.9799954295158386,
5
+ "eval_pearson": 0.7514687536537382,
6
+ "eval_runtime": 2.8106,
7
+ "eval_samples": 1500,
8
+ "eval_samples_per_second": 533.694,
9
+ "eval_spearmanr": 0.7471924680940966,
10
+ "eval_steps_per_second": 4.27,
11
+ "train_loss": 0.7678726120600625,
12
+ "train_runtime": 566.4946,
13
+ "train_samples": 5749,
14
+ "train_samples_per_second": 507.419,
15
+ "train_steps_per_second": 3.972
16
+ }
eval_results.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 14.0,
3
+ "eval_combined_score": 0.7493306108739174,
4
+ "eval_loss": 0.9799954295158386,
5
+ "eval_pearson": 0.7514687536537382,
6
+ "eval_runtime": 2.8106,
7
+ "eval_samples": 1500,
8
+ "eval_samples_per_second": 533.694,
9
+ "eval_spearmanr": 0.7471924680940966,
10
+ "eval_steps_per_second": 4.27
11
+ }
logs/events.out.tfevents.1686058598.serv-9207.2389173.20 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5654bfc0dab8975c9dbbeb97a34160295c8a55cdb56f66a4bd5a60b24bedc09c
3
+ size 473
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 14.0,
3
+ "train_loss": 0.7678726120600625,
4
+ "train_runtime": 566.4946,
5
+ "train_samples": 5749,
6
+ "train_samples_per_second": 507.419,
7
+ "train_steps_per_second": 3.972
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,263 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.9799954295158386,
3
+ "best_model_checkpoint": "hBERTv1_new_pretrain_w_init_48_stsb/checkpoint-405",
4
+ "epoch": 14.0,
5
+ "global_step": 630,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 1.0,
12
+ "learning_rate": 3.9200000000000004e-05,
13
+ "loss": 2.5456,
14
+ "step": 45
15
+ },
16
+ {
17
+ "epoch": 1.0,
18
+ "eval_combined_score": 0.11935926461991596,
19
+ "eval_loss": 2.270618438720703,
20
+ "eval_pearson": 0.12461202621384566,
21
+ "eval_runtime": 2.794,
22
+ "eval_samples_per_second": 536.862,
23
+ "eval_spearmanr": 0.11410650302598625,
24
+ "eval_steps_per_second": 4.295,
25
+ "step": 45
26
+ },
27
+ {
28
+ "epoch": 2.0,
29
+ "learning_rate": 3.8400000000000005e-05,
30
+ "loss": 2.0514,
31
+ "step": 90
32
+ },
33
+ {
34
+ "epoch": 2.0,
35
+ "eval_combined_score": 0.5231834246957023,
36
+ "eval_loss": 2.0612900257110596,
37
+ "eval_pearson": 0.5265750419588554,
38
+ "eval_runtime": 2.8043,
39
+ "eval_samples_per_second": 534.884,
40
+ "eval_spearmanr": 0.5197918074325493,
41
+ "eval_steps_per_second": 4.279,
42
+ "step": 90
43
+ },
44
+ {
45
+ "epoch": 3.0,
46
+ "learning_rate": 3.76e-05,
47
+ "loss": 1.3837,
48
+ "step": 135
49
+ },
50
+ {
51
+ "epoch": 3.0,
52
+ "eval_combined_score": 0.6897094834383319,
53
+ "eval_loss": 1.1984349489212036,
54
+ "eval_pearson": 0.685263316928048,
55
+ "eval_runtime": 2.8304,
56
+ "eval_samples_per_second": 529.967,
57
+ "eval_spearmanr": 0.694155649948616,
58
+ "eval_steps_per_second": 4.24,
59
+ "step": 135
60
+ },
61
+ {
62
+ "epoch": 4.0,
63
+ "learning_rate": 3.680000000000001e-05,
64
+ "loss": 1.0297,
65
+ "step": 180
66
+ },
67
+ {
68
+ "epoch": 4.0,
69
+ "eval_combined_score": 0.6914969217648269,
70
+ "eval_loss": 1.6176120042800903,
71
+ "eval_pearson": 0.6869258091116714,
72
+ "eval_runtime": 2.8207,
73
+ "eval_samples_per_second": 531.79,
74
+ "eval_spearmanr": 0.6960680344179824,
75
+ "eval_steps_per_second": 4.254,
76
+ "step": 180
77
+ },
78
+ {
79
+ "epoch": 5.0,
80
+ "learning_rate": 3.6e-05,
81
+ "loss": 0.8064,
82
+ "step": 225
83
+ },
84
+ {
85
+ "epoch": 5.0,
86
+ "eval_combined_score": 0.7460100236019249,
87
+ "eval_loss": 1.1443842649459839,
88
+ "eval_pearson": 0.7475621408974276,
89
+ "eval_runtime": 2.8189,
90
+ "eval_samples_per_second": 532.114,
91
+ "eval_spearmanr": 0.7444579063064223,
92
+ "eval_steps_per_second": 4.257,
93
+ "step": 225
94
+ },
95
+ {
96
+ "epoch": 6.0,
97
+ "learning_rate": 3.52e-05,
98
+ "loss": 0.604,
99
+ "step": 270
100
+ },
101
+ {
102
+ "epoch": 6.0,
103
+ "eval_combined_score": 0.7436314867221065,
104
+ "eval_loss": 1.2753756046295166,
105
+ "eval_pearson": 0.7422254000916392,
106
+ "eval_runtime": 2.8172,
107
+ "eval_samples_per_second": 532.449,
108
+ "eval_spearmanr": 0.7450375733525738,
109
+ "eval_steps_per_second": 4.26,
110
+ "step": 270
111
+ },
112
+ {
113
+ "epoch": 7.0,
114
+ "learning_rate": 3.44e-05,
115
+ "loss": 0.4818,
116
+ "step": 315
117
+ },
118
+ {
119
+ "epoch": 7.0,
120
+ "eval_combined_score": 0.7680210248146195,
121
+ "eval_loss": 1.1406958103179932,
122
+ "eval_pearson": 0.7687318203548877,
123
+ "eval_runtime": 2.8308,
124
+ "eval_samples_per_second": 529.887,
125
+ "eval_spearmanr": 0.7673102292743514,
126
+ "eval_steps_per_second": 4.239,
127
+ "step": 315
128
+ },
129
+ {
130
+ "epoch": 8.0,
131
+ "learning_rate": 3.3600000000000004e-05,
132
+ "loss": 0.3905,
133
+ "step": 360
134
+ },
135
+ {
136
+ "epoch": 8.0,
137
+ "eval_combined_score": 0.758227977038739,
138
+ "eval_loss": 1.1859891414642334,
139
+ "eval_pearson": 0.7560439754355848,
140
+ "eval_runtime": 2.8272,
141
+ "eval_samples_per_second": 530.555,
142
+ "eval_spearmanr": 0.7604119786418932,
143
+ "eval_steps_per_second": 4.244,
144
+ "step": 360
145
+ },
146
+ {
147
+ "epoch": 9.0,
148
+ "learning_rate": 3.28e-05,
149
+ "loss": 0.3476,
150
+ "step": 405
151
+ },
152
+ {
153
+ "epoch": 9.0,
154
+ "eval_combined_score": 0.7493306108739174,
155
+ "eval_loss": 0.9799954295158386,
156
+ "eval_pearson": 0.7514687536537382,
157
+ "eval_runtime": 2.8292,
158
+ "eval_samples_per_second": 530.193,
159
+ "eval_spearmanr": 0.7471924680940966,
160
+ "eval_steps_per_second": 4.242,
161
+ "step": 405
162
+ },
163
+ {
164
+ "epoch": 10.0,
165
+ "learning_rate": 3.2000000000000005e-05,
166
+ "loss": 0.2819,
167
+ "step": 450
168
+ },
169
+ {
170
+ "epoch": 10.0,
171
+ "eval_combined_score": 0.7514056262860007,
172
+ "eval_loss": 1.0156044960021973,
173
+ "eval_pearson": 0.7521360337612034,
174
+ "eval_runtime": 2.8179,
175
+ "eval_samples_per_second": 532.313,
176
+ "eval_spearmanr": 0.7506752188107981,
177
+ "eval_steps_per_second": 4.259,
178
+ "step": 450
179
+ },
180
+ {
181
+ "epoch": 11.0,
182
+ "learning_rate": 3.1200000000000006e-05,
183
+ "loss": 0.2418,
184
+ "step": 495
185
+ },
186
+ {
187
+ "epoch": 11.0,
188
+ "eval_combined_score": 0.749784975779181,
189
+ "eval_loss": 1.0174038410186768,
190
+ "eval_pearson": 0.7516160448200117,
191
+ "eval_runtime": 2.8194,
192
+ "eval_samples_per_second": 532.024,
193
+ "eval_spearmanr": 0.7479539067383503,
194
+ "eval_steps_per_second": 4.256,
195
+ "step": 495
196
+ },
197
+ {
198
+ "epoch": 12.0,
199
+ "learning_rate": 3.0400000000000004e-05,
200
+ "loss": 0.2068,
201
+ "step": 540
202
+ },
203
+ {
204
+ "epoch": 12.0,
205
+ "eval_combined_score": 0.7526651275891282,
206
+ "eval_loss": 1.236674427986145,
207
+ "eval_pearson": 0.7530062028197931,
208
+ "eval_runtime": 2.8215,
209
+ "eval_samples_per_second": 531.637,
210
+ "eval_spearmanr": 0.7523240523584632,
211
+ "eval_steps_per_second": 4.253,
212
+ "step": 540
213
+ },
214
+ {
215
+ "epoch": 13.0,
216
+ "learning_rate": 2.96e-05,
217
+ "loss": 0.1863,
218
+ "step": 585
219
+ },
220
+ {
221
+ "epoch": 13.0,
222
+ "eval_combined_score": 0.7479613454649323,
223
+ "eval_loss": 1.0072920322418213,
224
+ "eval_pearson": 0.7491318790965454,
225
+ "eval_runtime": 2.8345,
226
+ "eval_samples_per_second": 529.2,
227
+ "eval_spearmanr": 0.7467908118333191,
228
+ "eval_steps_per_second": 4.234,
229
+ "step": 585
230
+ },
231
+ {
232
+ "epoch": 14.0,
233
+ "learning_rate": 2.8800000000000002e-05,
234
+ "loss": 0.1929,
235
+ "step": 630
236
+ },
237
+ {
238
+ "epoch": 14.0,
239
+ "eval_combined_score": 0.7511028300281887,
240
+ "eval_loss": 1.0470010042190552,
241
+ "eval_pearson": 0.7516884780316463,
242
+ "eval_runtime": 2.8162,
243
+ "eval_samples_per_second": 532.63,
244
+ "eval_spearmanr": 0.750517182024731,
245
+ "eval_steps_per_second": 4.261,
246
+ "step": 630
247
+ },
248
+ {
249
+ "epoch": 14.0,
250
+ "step": 630,
251
+ "total_flos": 1.1919621946867712e+16,
252
+ "train_loss": 0.7678726120600625,
253
+ "train_runtime": 566.4946,
254
+ "train_samples_per_second": 507.419,
255
+ "train_steps_per_second": 3.972
256
+ }
257
+ ],
258
+ "max_steps": 2250,
259
+ "num_train_epochs": 50,
260
+ "total_flos": 1.1919621946867712e+16,
261
+ "trial_name": null,
262
+ "trial_params": null
263
+ }