Julien Chaumond commited on
Commit
cb8ef6d
1 Parent(s): 51cbb4b

save from trainer

Browse files
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ checkpoint*
README.md ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - text-classification
5
+ metrics:
6
+ - accuracy
7
+
8
+ model-index:
9
+ - name: model
10
+ results:
11
+ - task:
12
+ name: Text Classification
13
+ type: text-classification
14
+ metrics:
15
+ - name: Accuracy
16
+ type: accuracy
17
+ value: 0.2662102282047272
18
+ ---
19
+
20
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
21
+ should probably proofread and complete it, then remove this comment. -->
22
+
23
+ # model
24
+
25
+ This model is a fine-tuned version of [distilroberta-base](https://huggingface.co/distilroberta-base) on an unkown dataset.
26
+ It achieves the following results on the evaluation set:
27
+ - Loss: 2.9150
28
+ - Accuracy: 0.2662
29
+
30
+ ## Model description
31
+
32
+ More information needed
33
+
34
+ ## Intended uses & limitations
35
+
36
+ More information needed
37
+
38
+ ## Training and evaluation data
39
+
40
+ More information needed
41
+
42
+ ## Training procedure
43
+
44
+ ### Training hyperparameters
45
+
46
+ The following hyperparameters were used during training:
47
+ - learning_rate: 5e-05
48
+ - train_batch_size: 8
49
+ - eval_batch_size: 8
50
+ - seed: 42
51
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
52
+ - lr_scheduler_type: linear
53
+ - num_epochs: 3.0
54
+
55
+ ### Training results
56
+
57
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
58
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
59
+ | 3.0528 | 0.44 | 1000 | 3.0265 | 0.2223 |
60
+ | 2.9836 | 0.89 | 2000 | 2.9263 | 0.2332 |
61
+ | 2.7409 | 1.33 | 3000 | 2.9041 | 0.2533 |
62
+ | 2.7905 | 1.77 | 4000 | 2.8763 | 0.2606 |
63
+ | 2.4359 | 2.22 | 5000 | 2.9072 | 0.2642 |
64
+ | 2.4507 | 2.66 | 6000 | 2.9230 | 0.2644 |
65
+
66
+
67
+ ### Framework versions
68
+
69
+ - Transformers 4.7.0.dev0
70
+ - Pytorch 1.8.1+cu102
71
+ - Datasets 1.8.0
72
+ - Tokenizers 0.10.3
all_results.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.2662102282047272,
4
+ "eval_loss": 2.914989471435547,
5
+ "eval_runtime": 7.4525,
6
+ "eval_samples": 4966,
7
+ "eval_samples_per_second": 666.352,
8
+ "eval_steps_per_second": 83.328,
9
+ "train_loss": 2.764526012873939,
10
+ "train_runtime": 382.9395,
11
+ "train_samples": 18029,
12
+ "train_samples_per_second": 141.242,
13
+ "train_steps_per_second": 17.658
14
+ }
config.json ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilroberta-base",
3
+ "architectures": [
4
+ "RobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "eos_token_id": 2,
9
+ "gradient_checkpointing": false,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "id2label": {
14
+ "0": "agree",
15
+ "1": "applause",
16
+ "2": "awww",
17
+ "3": "dance",
18
+ "4": "deal_with_it",
19
+ "5": "do_not_want",
20
+ "6": "eww",
21
+ "7": "eye_roll",
22
+ "8": "facepalm",
23
+ "9": "fist_bump",
24
+ "10": "good_luck",
25
+ "11": "happy_dance",
26
+ "12": "hearts",
27
+ "13": "high_five",
28
+ "14": "hug",
29
+ "15": "idk",
30
+ "16": "kiss",
31
+ "17": "mic_drop",
32
+ "18": "no",
33
+ "19": "oh_snap",
34
+ "20": "ok",
35
+ "21": "omg",
36
+ "22": "oops",
37
+ "23": "please",
38
+ "24": "popcorn",
39
+ "25": "scared",
40
+ "26": "seriously",
41
+ "27": "shocked",
42
+ "28": "shrug",
43
+ "29": "sigh",
44
+ "30": "slow_clap",
45
+ "31": "smh",
46
+ "32": "sorry",
47
+ "33": "thank_you",
48
+ "34": "thumbs_down",
49
+ "35": "thumbs_up",
50
+ "36": "want",
51
+ "37": "win",
52
+ "38": "wink",
53
+ "39": "yawn",
54
+ "40": "yes",
55
+ "41": "yolo",
56
+ "42": "you_got_this"
57
+ },
58
+ "initializer_range": 0.02,
59
+ "intermediate_size": 3072,
60
+ "label2id": {
61
+ "agree": 0,
62
+ "applause": 1,
63
+ "awww": 2,
64
+ "dance": 3,
65
+ "deal_with_it": 4,
66
+ "do_not_want": 5,
67
+ "eww": 6,
68
+ "eye_roll": 7,
69
+ "facepalm": 8,
70
+ "fist_bump": 9,
71
+ "good_luck": 10,
72
+ "happy_dance": 11,
73
+ "hearts": 12,
74
+ "high_five": 13,
75
+ "hug": 14,
76
+ "idk": 15,
77
+ "kiss": 16,
78
+ "mic_drop": 17,
79
+ "no": 18,
80
+ "oh_snap": 19,
81
+ "ok": 20,
82
+ "omg": 21,
83
+ "oops": 22,
84
+ "please": 23,
85
+ "popcorn": 24,
86
+ "scared": 25,
87
+ "seriously": 26,
88
+ "shocked": 27,
89
+ "shrug": 28,
90
+ "sigh": 29,
91
+ "slow_clap": 30,
92
+ "smh": 31,
93
+ "sorry": 32,
94
+ "thank_you": 33,
95
+ "thumbs_down": 34,
96
+ "thumbs_up": 35,
97
+ "want": 36,
98
+ "win": 37,
99
+ "wink": 38,
100
+ "yawn": 39,
101
+ "yes": 40,
102
+ "yolo": 41,
103
+ "you_got_this": 42
104
+ },
105
+ "layer_norm_eps": 1e-05,
106
+ "max_position_embeddings": 514,
107
+ "model_type": "roberta",
108
+ "num_attention_heads": 12,
109
+ "num_hidden_layers": 6,
110
+ "pad_token_id": 1,
111
+ "position_embedding_type": "absolute",
112
+ "problem_type": "single_label_classification",
113
+ "transformers_version": "4.7.0.dev0",
114
+ "type_vocab_size": 1,
115
+ "use_cache": true,
116
+ "vocab_size": 50265
117
+ }
eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.2662102282047272,
4
+ "eval_loss": 2.914989471435547,
5
+ "eval_runtime": 7.4525,
6
+ "eval_samples": 4966,
7
+ "eval_samples_per_second": 666.352,
8
+ "eval_steps_per_second": 83.328
9
+ }
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:b54c4411f601d2c8693a27f20003a5860caa54e04de033a30314fcc24bb8a37a
3
+ size 328655081
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": "<unk>", "bos_token": "<s>", "eos_token": "</s>", "add_prefix_space": false, "errors": "replace", "sep_token": "</s>", "cls_token": "<s>", "pad_token": "<pad>", "mask_token": "<mask>", "model_max_length": 512, "special_tokens_map_file": null, "name_or_path": "distilroberta-base"}
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "train_loss": 2.764526012873939,
4
+ "train_runtime": 382.9395,
5
+ "train_samples": 18029,
6
+ "train_samples_per_second": 141.242,
7
+ "train_steps_per_second": 17.658
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,283 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "global_step": 6762,
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.0,
12
+ "learning_rate": 4.9992605737947355e-05,
13
+ "loss": 3.6847,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 0.09,
18
+ "learning_rate": 4.852114758947057e-05,
19
+ "loss": 3.4099,
20
+ "step": 200
21
+ },
22
+ {
23
+ "epoch": 0.18,
24
+ "learning_rate": 4.704229517894114e-05,
25
+ "loss": 3.2337,
26
+ "step": 400
27
+ },
28
+ {
29
+ "epoch": 0.27,
30
+ "learning_rate": 4.5563442768411715e-05,
31
+ "loss": 3.0788,
32
+ "step": 600
33
+ },
34
+ {
35
+ "epoch": 0.35,
36
+ "learning_rate": 4.408459035788228e-05,
37
+ "loss": 3.1394,
38
+ "step": 800
39
+ },
40
+ {
41
+ "epoch": 0.44,
42
+ "learning_rate": 4.260573794735286e-05,
43
+ "loss": 3.0528,
44
+ "step": 1000
45
+ },
46
+ {
47
+ "epoch": 0.44,
48
+ "eval_accuracy": 0.22231172025203705,
49
+ "eval_loss": 3.0265257358551025,
50
+ "eval_runtime": 7.2925,
51
+ "eval_samples_per_second": 680.975,
52
+ "eval_steps_per_second": 85.156,
53
+ "step": 1000
54
+ },
55
+ {
56
+ "epoch": 0.53,
57
+ "learning_rate": 4.112688553682343e-05,
58
+ "loss": 2.9704,
59
+ "step": 1200
60
+ },
61
+ {
62
+ "epoch": 0.62,
63
+ "learning_rate": 3.9648033126293996e-05,
64
+ "loss": 3.0544,
65
+ "step": 1400
66
+ },
67
+ {
68
+ "epoch": 0.71,
69
+ "learning_rate": 3.816918071576457e-05,
70
+ "loss": 2.9979,
71
+ "step": 1600
72
+ },
73
+ {
74
+ "epoch": 0.8,
75
+ "learning_rate": 3.669032830523514e-05,
76
+ "loss": 3.0237,
77
+ "step": 1800
78
+ },
79
+ {
80
+ "epoch": 0.89,
81
+ "learning_rate": 3.521147589470571e-05,
82
+ "loss": 2.9836,
83
+ "step": 2000
84
+ },
85
+ {
86
+ "epoch": 0.89,
87
+ "eval_accuracy": 0.23318566381931305,
88
+ "eval_loss": 2.926269054412842,
89
+ "eval_runtime": 7.3605,
90
+ "eval_samples_per_second": 674.678,
91
+ "eval_steps_per_second": 84.369,
92
+ "step": 2000
93
+ },
94
+ {
95
+ "epoch": 0.98,
96
+ "learning_rate": 3.3732623484176276e-05,
97
+ "loss": 2.9896,
98
+ "step": 2200
99
+ },
100
+ {
101
+ "epoch": 1.06,
102
+ "learning_rate": 3.225377107364685e-05,
103
+ "loss": 2.8578,
104
+ "step": 2400
105
+ },
106
+ {
107
+ "epoch": 1.15,
108
+ "learning_rate": 3.077491866311743e-05,
109
+ "loss": 2.7667,
110
+ "step": 2600
111
+ },
112
+ {
113
+ "epoch": 1.24,
114
+ "learning_rate": 2.9296066252587996e-05,
115
+ "loss": 2.8017,
116
+ "step": 2800
117
+ },
118
+ {
119
+ "epoch": 1.33,
120
+ "learning_rate": 2.7817213842058564e-05,
121
+ "loss": 2.7409,
122
+ "step": 3000
123
+ },
124
+ {
125
+ "epoch": 1.33,
126
+ "eval_accuracy": 0.2533226013183594,
127
+ "eval_loss": 2.9040770530700684,
128
+ "eval_runtime": 7.3851,
129
+ "eval_samples_per_second": 672.437,
130
+ "eval_steps_per_second": 84.088,
131
+ "step": 3000
132
+ },
133
+ {
134
+ "epoch": 1.42,
135
+ "learning_rate": 2.6338361431529136e-05,
136
+ "loss": 2.7605,
137
+ "step": 3200
138
+ },
139
+ {
140
+ "epoch": 1.51,
141
+ "learning_rate": 2.4859509020999705e-05,
142
+ "loss": 2.7569,
143
+ "step": 3400
144
+ },
145
+ {
146
+ "epoch": 1.6,
147
+ "learning_rate": 2.3380656610470276e-05,
148
+ "loss": 2.7154,
149
+ "step": 3600
150
+ },
151
+ {
152
+ "epoch": 1.69,
153
+ "learning_rate": 2.190180419994085e-05,
154
+ "loss": 2.788,
155
+ "step": 3800
156
+ },
157
+ {
158
+ "epoch": 1.77,
159
+ "learning_rate": 2.0422951789411417e-05,
160
+ "loss": 2.7905,
161
+ "step": 4000
162
+ },
163
+ {
164
+ "epoch": 1.77,
165
+ "eval_accuracy": 0.2605718970298767,
166
+ "eval_loss": 2.876274824142456,
167
+ "eval_runtime": 7.4268,
168
+ "eval_samples_per_second": 668.66,
169
+ "eval_steps_per_second": 83.616,
170
+ "step": 4000
171
+ },
172
+ {
173
+ "epoch": 1.86,
174
+ "learning_rate": 1.894409937888199e-05,
175
+ "loss": 2.7044,
176
+ "step": 4200
177
+ },
178
+ {
179
+ "epoch": 1.95,
180
+ "learning_rate": 1.7465246968352557e-05,
181
+ "loss": 2.7526,
182
+ "step": 4400
183
+ },
184
+ {
185
+ "epoch": 2.04,
186
+ "learning_rate": 1.5986394557823133e-05,
187
+ "loss": 2.6027,
188
+ "step": 4600
189
+ },
190
+ {
191
+ "epoch": 2.13,
192
+ "learning_rate": 1.4507542147293701e-05,
193
+ "loss": 2.4391,
194
+ "step": 4800
195
+ },
196
+ {
197
+ "epoch": 2.22,
198
+ "learning_rate": 1.3028689736764271e-05,
199
+ "loss": 2.4359,
200
+ "step": 5000
201
+ },
202
+ {
203
+ "epoch": 2.22,
204
+ "eval_accuracy": 0.2641965448856354,
205
+ "eval_loss": 2.9071688652038574,
206
+ "eval_runtime": 7.4381,
207
+ "eval_samples_per_second": 667.644,
208
+ "eval_steps_per_second": 83.489,
209
+ "step": 5000
210
+ },
211
+ {
212
+ "epoch": 2.31,
213
+ "learning_rate": 1.1549837326234843e-05,
214
+ "loss": 2.4514,
215
+ "step": 5200
216
+ },
217
+ {
218
+ "epoch": 2.4,
219
+ "learning_rate": 1.0070984915705414e-05,
220
+ "loss": 2.4754,
221
+ "step": 5400
222
+ },
223
+ {
224
+ "epoch": 2.48,
225
+ "learning_rate": 8.592132505175984e-06,
226
+ "loss": 2.4698,
227
+ "step": 5600
228
+ },
229
+ {
230
+ "epoch": 2.57,
231
+ "learning_rate": 7.113280094646555e-06,
232
+ "loss": 2.4732,
233
+ "step": 5800
234
+ },
235
+ {
236
+ "epoch": 2.66,
237
+ "learning_rate": 5.634427684117125e-06,
238
+ "loss": 2.4507,
239
+ "step": 6000
240
+ },
241
+ {
242
+ "epoch": 2.66,
243
+ "eval_accuracy": 0.26439791917800903,
244
+ "eval_loss": 2.9230332374572754,
245
+ "eval_runtime": 7.4558,
246
+ "eval_samples_per_second": 666.058,
247
+ "eval_steps_per_second": 83.291,
248
+ "step": 6000
249
+ },
250
+ {
251
+ "epoch": 2.75,
252
+ "learning_rate": 4.155575273587696e-06,
253
+ "loss": 2.5628,
254
+ "step": 6200
255
+ },
256
+ {
257
+ "epoch": 2.84,
258
+ "learning_rate": 2.6767228630582667e-06,
259
+ "loss": 2.3556,
260
+ "step": 6400
261
+ },
262
+ {
263
+ "epoch": 2.93,
264
+ "learning_rate": 1.1978704525288378e-06,
265
+ "loss": 2.4206,
266
+ "step": 6600
267
+ },
268
+ {
269
+ "epoch": 3.0,
270
+ "step": 6762,
271
+ "total_flos": 3412474671843072.0,
272
+ "train_loss": 2.764526012873939,
273
+ "train_runtime": 382.9395,
274
+ "train_samples_per_second": 141.242,
275
+ "train_steps_per_second": 17.658
276
+ }
277
+ ],
278
+ "max_steps": 6762,
279
+ "num_train_epochs": 3,
280
+ "total_flos": 3412474671843072.0,
281
+ "trial_name": null,
282
+ "trial_params": null
283
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5740560245546a0993ff4a9a2d710272d28f10f294b413d89bac9645d1e8369c
3
+ size 2415
vocab.json ADDED
The diff for this file is too large to render. See raw diff