mjtwins commited on
Commit
b4086e8
1 Parent(s): 3921a2f

model upload

Browse files
added_tokens.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"[E]": 50266, "[S]": 50265}
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/hdd1/mujeen/retrieval_prf/output/labeler_multi.train.v4.1_nq.dev.v4.1_rlmulti_title",
3
+ "architectures": [
4
+ "RobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 1024,
13
+ "id2label": {
14
+ "0": 0,
15
+ "1": 1
16
+ },
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 4096,
19
+ "label2id": {
20
+ "0": 0,
21
+ "1": 1
22
+ },
23
+ "layer_norm_eps": 1e-05,
24
+ "max_position_embeddings": 514,
25
+ "model_type": "roberta",
26
+ "num_attention_heads": 16,
27
+ "num_hidden_layers": 24,
28
+ "pad_token_id": 1,
29
+ "position_embedding_type": "absolute",
30
+ "problem_type": "single_label_classification",
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.13.0",
33
+ "type_vocab_size": 1,
34
+ "use_cache": true,
35
+ "vocab_size": 50267
36
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f48a9a5ed769ae57f3a1d4de834f0a333f31e8009667c8c62722dab9fff9d509
3
+ size 1421619501
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"bos_token": "<s>", "eos_token": "</s>", "unk_token": "<unk>", "sep_token": "</s>", "pad_token": "<pad>", "cls_token": "<s>", "mask_token": {"content": "<mask>", "single_word": false, "lstrip": true, "rstrip": false, "normalized": false}}
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"unk_token": {"content": "<unk>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "bos_token": {"content": "<s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "eos_token": {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "add_prefix_space": false, "errors": "replace", "sep_token": {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "cls_token": {"content": "<s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "pad_token": {"content": "<pad>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "mask_token": {"content": "<mask>", "single_word": false, "lstrip": true, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "do_lower_case": false, "model_max_length": 512, "special_tokens_map_file": "/mnt/hdd-nfs/jungsoo/DensePhrases/resources/ckpts/roberta-large-multi-lr_2e-5/special_tokens_map.json", "name_or_path": "/hdd1/mujeen/retrieval_prf/output/labeler_multi.train.v4.1_nq.dev.v4.1_rlmulti_title", "tokenizer_class": "RobertaTokenizer"}
trainer_state.json ADDED
@@ -0,0 +1,325 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.8122448979591836,
3
+ "best_model_checkpoint": "/hdd1/mujeen/retrieval_prf/output/labeler_multi.train.v4.1_nq.dev.v4.1_rlmulti_title_wq/checkpoint-300",
4
+ "epoch": 1.0,
5
+ "global_step": 3080,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.05,
12
+ "learning_rate": 1.903896103896104e-05,
13
+ "loss": 0.2844,
14
+ "step": 150
15
+ },
16
+ {
17
+ "epoch": 0.05,
18
+ "eval_accuracy": 0.7714285714285715,
19
+ "eval_loss": 0.5584465861320496,
20
+ "eval_runtime": 3.5617,
21
+ "eval_samples_per_second": 137.576,
22
+ "eval_steps_per_second": 17.408,
23
+ "step": 150
24
+ },
25
+ {
26
+ "epoch": 0.1,
27
+ "learning_rate": 1.8064935064935067e-05,
28
+ "loss": 0.2673,
29
+ "step": 300
30
+ },
31
+ {
32
+ "epoch": 0.1,
33
+ "eval_accuracy": 0.8122448979591836,
34
+ "eval_loss": 0.5982859134674072,
35
+ "eval_runtime": 3.1189,
36
+ "eval_samples_per_second": 157.106,
37
+ "eval_steps_per_second": 19.879,
38
+ "step": 300
39
+ },
40
+ {
41
+ "epoch": 0.15,
42
+ "learning_rate": 1.7090909090909092e-05,
43
+ "loss": 0.2653,
44
+ "step": 450
45
+ },
46
+ {
47
+ "epoch": 0.15,
48
+ "eval_accuracy": 0.8040816326530612,
49
+ "eval_loss": 0.5716066956520081,
50
+ "eval_runtime": 3.0903,
51
+ "eval_samples_per_second": 158.56,
52
+ "eval_steps_per_second": 20.063,
53
+ "step": 450
54
+ },
55
+ {
56
+ "epoch": 0.19,
57
+ "learning_rate": 1.6116883116883118e-05,
58
+ "loss": 0.2453,
59
+ "step": 600
60
+ },
61
+ {
62
+ "epoch": 0.19,
63
+ "eval_accuracy": 0.7836734693877551,
64
+ "eval_loss": 0.6890708208084106,
65
+ "eval_runtime": 2.9422,
66
+ "eval_samples_per_second": 166.541,
67
+ "eval_steps_per_second": 21.073,
68
+ "step": 600
69
+ },
70
+ {
71
+ "epoch": 0.24,
72
+ "learning_rate": 1.514935064935065e-05,
73
+ "loss": 0.2234,
74
+ "step": 750
75
+ },
76
+ {
77
+ "epoch": 0.24,
78
+ "eval_accuracy": 0.8,
79
+ "eval_loss": 0.6545684337615967,
80
+ "eval_runtime": 3.0903,
81
+ "eval_samples_per_second": 158.559,
82
+ "eval_steps_per_second": 20.063,
83
+ "step": 750
84
+ },
85
+ {
86
+ "epoch": 0.29,
87
+ "learning_rate": 1.4175324675324675e-05,
88
+ "loss": 0.2117,
89
+ "step": 900
90
+ },
91
+ {
92
+ "epoch": 0.29,
93
+ "eval_accuracy": 0.7938775510204081,
94
+ "eval_loss": 0.6074094772338867,
95
+ "eval_runtime": 3.4718,
96
+ "eval_samples_per_second": 141.137,
97
+ "eval_steps_per_second": 17.858,
98
+ "step": 900
99
+ },
100
+ {
101
+ "epoch": 0.34,
102
+ "learning_rate": 1.3201298701298702e-05,
103
+ "loss": 0.1867,
104
+ "step": 1050
105
+ },
106
+ {
107
+ "epoch": 0.34,
108
+ "eval_accuracy": 0.7938775510204081,
109
+ "eval_loss": 0.7075100541114807,
110
+ "eval_runtime": 3.2481,
111
+ "eval_samples_per_second": 150.857,
112
+ "eval_steps_per_second": 19.088,
113
+ "step": 1050
114
+ },
115
+ {
116
+ "epoch": 0.39,
117
+ "learning_rate": 1.2227272727272728e-05,
118
+ "loss": 0.179,
119
+ "step": 1200
120
+ },
121
+ {
122
+ "epoch": 0.39,
123
+ "eval_accuracy": 0.8040816326530612,
124
+ "eval_loss": 0.6317723989486694,
125
+ "eval_runtime": 3.0265,
126
+ "eval_samples_per_second": 161.903,
127
+ "eval_steps_per_second": 20.486,
128
+ "step": 1200
129
+ },
130
+ {
131
+ "epoch": 0.44,
132
+ "learning_rate": 1.1253246753246754e-05,
133
+ "loss": 0.1718,
134
+ "step": 1350
135
+ },
136
+ {
137
+ "epoch": 0.44,
138
+ "eval_accuracy": 0.7836734693877551,
139
+ "eval_loss": 0.7868985533714294,
140
+ "eval_runtime": 3.0224,
141
+ "eval_samples_per_second": 162.125,
142
+ "eval_steps_per_second": 20.514,
143
+ "step": 1350
144
+ },
145
+ {
146
+ "epoch": 0.49,
147
+ "learning_rate": 1.027922077922078e-05,
148
+ "loss": 0.1713,
149
+ "step": 1500
150
+ },
151
+ {
152
+ "epoch": 0.49,
153
+ "eval_accuracy": 0.7775510204081633,
154
+ "eval_loss": 0.6507557034492493,
155
+ "eval_runtime": 3.6832,
156
+ "eval_samples_per_second": 133.036,
157
+ "eval_steps_per_second": 16.833,
158
+ "step": 1500
159
+ },
160
+ {
161
+ "epoch": 0.54,
162
+ "learning_rate": 9.318181818181819e-06,
163
+ "loss": 0.1502,
164
+ "step": 1650
165
+ },
166
+ {
167
+ "epoch": 0.54,
168
+ "eval_accuracy": 0.7836734693877551,
169
+ "eval_loss": 0.738645613193512,
170
+ "eval_runtime": 3.2984,
171
+ "eval_samples_per_second": 148.555,
172
+ "eval_steps_per_second": 18.797,
173
+ "step": 1650
174
+ },
175
+ {
176
+ "epoch": 0.58,
177
+ "learning_rate": 8.344155844155845e-06,
178
+ "loss": 0.1541,
179
+ "step": 1800
180
+ },
181
+ {
182
+ "epoch": 0.58,
183
+ "eval_accuracy": 0.7877551020408163,
184
+ "eval_loss": 0.8032358884811401,
185
+ "eval_runtime": 3.6265,
186
+ "eval_samples_per_second": 135.118,
187
+ "eval_steps_per_second": 17.097,
188
+ "step": 1800
189
+ },
190
+ {
191
+ "epoch": 0.63,
192
+ "learning_rate": 7.370129870129871e-06,
193
+ "loss": 0.1358,
194
+ "step": 1950
195
+ },
196
+ {
197
+ "epoch": 0.63,
198
+ "eval_accuracy": 0.7959183673469388,
199
+ "eval_loss": 0.7314993739128113,
200
+ "eval_runtime": 3.7106,
201
+ "eval_samples_per_second": 132.054,
202
+ "eval_steps_per_second": 16.709,
203
+ "step": 1950
204
+ },
205
+ {
206
+ "epoch": 0.68,
207
+ "learning_rate": 6.3961038961038964e-06,
208
+ "loss": 0.1251,
209
+ "step": 2100
210
+ },
211
+ {
212
+ "epoch": 0.68,
213
+ "eval_accuracy": 0.7816326530612245,
214
+ "eval_loss": 0.9869228601455688,
215
+ "eval_runtime": 3.3927,
216
+ "eval_samples_per_second": 144.427,
217
+ "eval_steps_per_second": 18.274,
218
+ "step": 2100
219
+ },
220
+ {
221
+ "epoch": 0.73,
222
+ "learning_rate": 5.422077922077923e-06,
223
+ "loss": 0.1244,
224
+ "step": 2250
225
+ },
226
+ {
227
+ "epoch": 0.73,
228
+ "eval_accuracy": 0.7836734693877551,
229
+ "eval_loss": 0.8370808362960815,
230
+ "eval_runtime": 3.5148,
231
+ "eval_samples_per_second": 139.41,
232
+ "eval_steps_per_second": 17.64,
233
+ "step": 2250
234
+ },
235
+ {
236
+ "epoch": 0.78,
237
+ "learning_rate": 4.448051948051948e-06,
238
+ "loss": 0.1223,
239
+ "step": 2400
240
+ },
241
+ {
242
+ "epoch": 0.78,
243
+ "eval_accuracy": 0.7816326530612245,
244
+ "eval_loss": 0.833113968372345,
245
+ "eval_runtime": 3.1172,
246
+ "eval_samples_per_second": 157.192,
247
+ "eval_steps_per_second": 19.89,
248
+ "step": 2400
249
+ },
250
+ {
251
+ "epoch": 0.83,
252
+ "learning_rate": 3.474025974025974e-06,
253
+ "loss": 0.096,
254
+ "step": 2550
255
+ },
256
+ {
257
+ "epoch": 0.83,
258
+ "eval_accuracy": 0.7979591836734694,
259
+ "eval_loss": 0.8976467847824097,
260
+ "eval_runtime": 3.0168,
261
+ "eval_samples_per_second": 162.426,
262
+ "eval_steps_per_second": 20.552,
263
+ "step": 2550
264
+ },
265
+ {
266
+ "epoch": 0.88,
267
+ "learning_rate": 2.5e-06,
268
+ "loss": 0.1113,
269
+ "step": 2700
270
+ },
271
+ {
272
+ "epoch": 0.88,
273
+ "eval_accuracy": 0.7755102040816326,
274
+ "eval_loss": 0.96781325340271,
275
+ "eval_runtime": 3.3837,
276
+ "eval_samples_per_second": 144.814,
277
+ "eval_steps_per_second": 18.323,
278
+ "step": 2700
279
+ },
280
+ {
281
+ "epoch": 0.93,
282
+ "learning_rate": 1.525974025974026e-06,
283
+ "loss": 0.094,
284
+ "step": 2850
285
+ },
286
+ {
287
+ "epoch": 0.93,
288
+ "eval_accuracy": 0.7816326530612245,
289
+ "eval_loss": 1.0002238750457764,
290
+ "eval_runtime": 2.9984,
291
+ "eval_samples_per_second": 163.421,
292
+ "eval_steps_per_second": 20.678,
293
+ "step": 2850
294
+ },
295
+ {
296
+ "epoch": 0.97,
297
+ "learning_rate": 5.51948051948052e-07,
298
+ "loss": 0.1084,
299
+ "step": 3000
300
+ },
301
+ {
302
+ "epoch": 0.97,
303
+ "eval_accuracy": 0.7836734693877551,
304
+ "eval_loss": 1.0199074745178223,
305
+ "eval_runtime": 3.3069,
306
+ "eval_samples_per_second": 148.174,
307
+ "eval_steps_per_second": 18.748,
308
+ "step": 3000
309
+ },
310
+ {
311
+ "epoch": 1.0,
312
+ "step": 3080,
313
+ "total_flos": 2.867459611823923e+16,
314
+ "train_loss": 0.16962762838834292,
315
+ "train_runtime": 2524.7018,
316
+ "train_samples_per_second": 39.038,
317
+ "train_steps_per_second": 1.22
318
+ }
319
+ ],
320
+ "max_steps": 3080,
321
+ "num_train_epochs": 1,
322
+ "total_flos": 2.867459611823923e+16,
323
+ "trial_name": null,
324
+ "trial_params": null
325
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6019fb427b7a72f0d6d3d0e64cdb6f3800011d927b67053ea58255ba235a4a3b
3
+ size 3055
vocab.json ADDED
The diff for this file is too large to render. See raw diff