Upload folder using huggingface_hub
Browse files- checkpoint-1648/config.json +44 -0
- checkpoint-1648/model.safetensors +3 -0
- checkpoint-1648/optimizer.pt +3 -0
- checkpoint-1648/rng_state.pth +3 -0
- checkpoint-1648/scheduler.pt +3 -0
- checkpoint-1648/trainer_state.json +166 -0
- checkpoint-1648/training_args.bin +3 -0
- checkpoint-3296/config.json +44 -0
- checkpoint-3296/model.safetensors +3 -0
- checkpoint-3296/optimizer.pt +3 -0
- checkpoint-3296/rng_state.pth +3 -0
- checkpoint-3296/scheduler.pt +3 -0
- checkpoint-3296/trainer_state.json +290 -0
- checkpoint-3296/training_args.bin +3 -0
- checkpoint-4944/config.json +44 -0
- checkpoint-4944/model.safetensors +3 -0
- checkpoint-4944/optimizer.pt +3 -0
- checkpoint-4944/rng_state.pth +3 -0
- checkpoint-4944/scheduler.pt +3 -0
- checkpoint-4944/trainer_state.json +421 -0
- checkpoint-4944/training_args.bin +3 -0
- config.json +44 -0
- confusion_matrix.png +0 -0
- model.safetensors +3 -0
- special_tokens_map.json +7 -0
- tokenizer.json +0 -0
- tokenizer_config.json +56 -0
- training_args.bin +3 -0
- type_mapping.json +1 -0
- vocab.txt +0 -0
checkpoint-1648/config.json
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "distilbert-base-multilingual-cased",
|
3 |
+
"activation": "gelu",
|
4 |
+
"architectures": [
|
5 |
+
"DistilBertForSequenceClassification"
|
6 |
+
],
|
7 |
+
"attention_dropout": 0.1,
|
8 |
+
"dim": 768,
|
9 |
+
"dropout": 0.1,
|
10 |
+
"hidden_dim": 3072,
|
11 |
+
"id2label": {
|
12 |
+
"0": "LABEL_0",
|
13 |
+
"1": "LABEL_1",
|
14 |
+
"2": "LABEL_2",
|
15 |
+
"3": "LABEL_3",
|
16 |
+
"4": "LABEL_4",
|
17 |
+
"5": "LABEL_5",
|
18 |
+
"6": "LABEL_6"
|
19 |
+
},
|
20 |
+
"initializer_range": 0.02,
|
21 |
+
"label2id": {
|
22 |
+
"LABEL_0": 0,
|
23 |
+
"LABEL_1": 1,
|
24 |
+
"LABEL_2": 2,
|
25 |
+
"LABEL_3": 3,
|
26 |
+
"LABEL_4": 4,
|
27 |
+
"LABEL_5": 5,
|
28 |
+
"LABEL_6": 6
|
29 |
+
},
|
30 |
+
"max_position_embeddings": 512,
|
31 |
+
"model_type": "distilbert",
|
32 |
+
"n_heads": 12,
|
33 |
+
"n_layers": 6,
|
34 |
+
"output_past": true,
|
35 |
+
"pad_token_id": 0,
|
36 |
+
"problem_type": "single_label_classification",
|
37 |
+
"qa_dropout": 0.1,
|
38 |
+
"seq_classif_dropout": 0.2,
|
39 |
+
"sinusoidal_pos_embds": false,
|
40 |
+
"tie_weights_": true,
|
41 |
+
"torch_dtype": "float32",
|
42 |
+
"transformers_version": "4.49.0",
|
43 |
+
"vocab_size": 119547
|
44 |
+
}
|
checkpoint-1648/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b127042ca9191b532a59773b581d6bd536541118bc87bd2d236e7180ef11bf42
|
3 |
+
size 541332756
|
checkpoint-1648/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bb0d2b238e0e306d6873e263a46e497e316883004aa945dbd0dbe5b3cf44d870
|
3 |
+
size 1082727610
|
checkpoint-1648/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3ba3b48e31073f7530281564c9ca8cea953948e8e36bf40e284519532add5360
|
3 |
+
size 14244
|
checkpoint-1648/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:21d4589dc98af400cfd4548dd7f95e09263e025d27f9d116be6ea3a835592735
|
3 |
+
size 1064
|
checkpoint-1648/trainer_state.json
ADDED
@@ -0,0 +1,166 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": 0.87963672068735,
|
3 |
+
"best_model_checkpoint": "distilbert-type-classifier/checkpoint-1648",
|
4 |
+
"epoch": 1.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 1648,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.06067961165048544,
|
13 |
+
"grad_norm": 7.083491325378418,
|
14 |
+
"learning_rate": 1.959546925566343e-05,
|
15 |
+
"loss": 1.3291,
|
16 |
+
"step": 100
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.12135922330097088,
|
20 |
+
"grad_norm": 5.435772895812988,
|
21 |
+
"learning_rate": 1.919093851132686e-05,
|
22 |
+
"loss": 0.7948,
|
23 |
+
"step": 200
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.1820388349514563,
|
27 |
+
"grad_norm": 5.531534671783447,
|
28 |
+
"learning_rate": 1.8786407766990295e-05,
|
29 |
+
"loss": 0.59,
|
30 |
+
"step": 300
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.24271844660194175,
|
34 |
+
"grad_norm": 5.0550031661987305,
|
35 |
+
"learning_rate": 1.838187702265372e-05,
|
36 |
+
"loss": 0.5483,
|
37 |
+
"step": 400
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.30339805825242716,
|
41 |
+
"grad_norm": 9.326889038085938,
|
42 |
+
"learning_rate": 1.7977346278317155e-05,
|
43 |
+
"loss": 0.472,
|
44 |
+
"step": 500
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.3640776699029126,
|
48 |
+
"grad_norm": 11.223947525024414,
|
49 |
+
"learning_rate": 1.7572815533980585e-05,
|
50 |
+
"loss": 0.4334,
|
51 |
+
"step": 600
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.42475728155339804,
|
55 |
+
"grad_norm": 6.310647487640381,
|
56 |
+
"learning_rate": 1.7168284789644015e-05,
|
57 |
+
"loss": 0.3993,
|
58 |
+
"step": 700
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.4854368932038835,
|
62 |
+
"grad_norm": 13.495203018188477,
|
63 |
+
"learning_rate": 1.6763754045307445e-05,
|
64 |
+
"loss": 0.3976,
|
65 |
+
"step": 800
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.5461165048543689,
|
69 |
+
"grad_norm": 9.985793113708496,
|
70 |
+
"learning_rate": 1.6359223300970874e-05,
|
71 |
+
"loss": 0.4045,
|
72 |
+
"step": 900
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.6067961165048543,
|
76 |
+
"grad_norm": 5.920831203460693,
|
77 |
+
"learning_rate": 1.5954692556634304e-05,
|
78 |
+
"loss": 0.3661,
|
79 |
+
"step": 1000
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.6674757281553398,
|
83 |
+
"grad_norm": 7.346311092376709,
|
84 |
+
"learning_rate": 1.5550161812297734e-05,
|
85 |
+
"loss": 0.3527,
|
86 |
+
"step": 1100
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.7281553398058253,
|
90 |
+
"grad_norm": 7.223079681396484,
|
91 |
+
"learning_rate": 1.5145631067961166e-05,
|
92 |
+
"loss": 0.361,
|
93 |
+
"step": 1200
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.7888349514563107,
|
97 |
+
"grad_norm": 5.1326799392700195,
|
98 |
+
"learning_rate": 1.4741100323624598e-05,
|
99 |
+
"loss": 0.3892,
|
100 |
+
"step": 1300
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.8495145631067961,
|
104 |
+
"grad_norm": 7.782454967498779,
|
105 |
+
"learning_rate": 1.4336569579288026e-05,
|
106 |
+
"loss": 0.3502,
|
107 |
+
"step": 1400
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.9101941747572816,
|
111 |
+
"grad_norm": 8.951793670654297,
|
112 |
+
"learning_rate": 1.3932038834951458e-05,
|
113 |
+
"loss": 0.2934,
|
114 |
+
"step": 1500
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.970873786407767,
|
118 |
+
"grad_norm": 9.385202407836914,
|
119 |
+
"learning_rate": 1.3527508090614887e-05,
|
120 |
+
"loss": 0.3436,
|
121 |
+
"step": 1600
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 1.0,
|
125 |
+
"eval_accuracy": 0.8803116147308782,
|
126 |
+
"eval_f1": 0.87963672068735,
|
127 |
+
"eval_loss": 0.30396342277526855,
|
128 |
+
"eval_precision": 0.8827457335389428,
|
129 |
+
"eval_recall": 0.8803116147308782,
|
130 |
+
"eval_runtime": 21.997,
|
131 |
+
"eval_samples_per_second": 256.762,
|
132 |
+
"eval_steps_per_second": 16.048,
|
133 |
+
"step": 1648
|
134 |
+
}
|
135 |
+
],
|
136 |
+
"logging_steps": 100,
|
137 |
+
"max_steps": 4944,
|
138 |
+
"num_input_tokens_seen": 0,
|
139 |
+
"num_train_epochs": 3,
|
140 |
+
"save_steps": 500,
|
141 |
+
"stateful_callbacks": {
|
142 |
+
"EarlyStoppingCallback": {
|
143 |
+
"args": {
|
144 |
+
"early_stopping_patience": 2,
|
145 |
+
"early_stopping_threshold": 0.0
|
146 |
+
},
|
147 |
+
"attributes": {
|
148 |
+
"early_stopping_patience_counter": 0
|
149 |
+
}
|
150 |
+
},
|
151 |
+
"TrainerControl": {
|
152 |
+
"args": {
|
153 |
+
"should_epoch_stop": false,
|
154 |
+
"should_evaluate": false,
|
155 |
+
"should_log": false,
|
156 |
+
"should_save": true,
|
157 |
+
"should_training_stop": false
|
158 |
+
},
|
159 |
+
"attributes": {}
|
160 |
+
}
|
161 |
+
},
|
162 |
+
"total_flos": 872905517958144.0,
|
163 |
+
"train_batch_size": 16,
|
164 |
+
"trial_name": null,
|
165 |
+
"trial_params": null
|
166 |
+
}
|
checkpoint-1648/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:609d9e093ff10cb72ff2ceab0b60b959ffb5a8e69641d5f5afc035f09f45f51b
|
3 |
+
size 5304
|
checkpoint-3296/config.json
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "distilbert-base-multilingual-cased",
|
3 |
+
"activation": "gelu",
|
4 |
+
"architectures": [
|
5 |
+
"DistilBertForSequenceClassification"
|
6 |
+
],
|
7 |
+
"attention_dropout": 0.1,
|
8 |
+
"dim": 768,
|
9 |
+
"dropout": 0.1,
|
10 |
+
"hidden_dim": 3072,
|
11 |
+
"id2label": {
|
12 |
+
"0": "LABEL_0",
|
13 |
+
"1": "LABEL_1",
|
14 |
+
"2": "LABEL_2",
|
15 |
+
"3": "LABEL_3",
|
16 |
+
"4": "LABEL_4",
|
17 |
+
"5": "LABEL_5",
|
18 |
+
"6": "LABEL_6"
|
19 |
+
},
|
20 |
+
"initializer_range": 0.02,
|
21 |
+
"label2id": {
|
22 |
+
"LABEL_0": 0,
|
23 |
+
"LABEL_1": 1,
|
24 |
+
"LABEL_2": 2,
|
25 |
+
"LABEL_3": 3,
|
26 |
+
"LABEL_4": 4,
|
27 |
+
"LABEL_5": 5,
|
28 |
+
"LABEL_6": 6
|
29 |
+
},
|
30 |
+
"max_position_embeddings": 512,
|
31 |
+
"model_type": "distilbert",
|
32 |
+
"n_heads": 12,
|
33 |
+
"n_layers": 6,
|
34 |
+
"output_past": true,
|
35 |
+
"pad_token_id": 0,
|
36 |
+
"problem_type": "single_label_classification",
|
37 |
+
"qa_dropout": 0.1,
|
38 |
+
"seq_classif_dropout": 0.2,
|
39 |
+
"sinusoidal_pos_embds": false,
|
40 |
+
"tie_weights_": true,
|
41 |
+
"torch_dtype": "float32",
|
42 |
+
"transformers_version": "4.49.0",
|
43 |
+
"vocab_size": 119547
|
44 |
+
}
|
checkpoint-3296/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3e5bfcf35f0a247d68dd865628094b6058cc18b89924f880ab9417eb04731ce5
|
3 |
+
size 541332756
|
checkpoint-3296/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8182b3aea9a6c91bd337d14cfa4a502f5e40c5aca604a463ff7331f80862faf1
|
3 |
+
size 1082727610
|
checkpoint-3296/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ec96d0b8b5f52722b8b82f029bdb73b1ee05b7d00a36093eeedec1c70c5621f2
|
3 |
+
size 14244
|
checkpoint-3296/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dc9576b6ded02fee58504ff02bb633c9ed789e6a4b3782812ebca632813095fe
|
3 |
+
size 1064
|
checkpoint-3296/trainer_state.json
ADDED
@@ -0,0 +1,290 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": 0.8996119975479232,
|
3 |
+
"best_model_checkpoint": "distilbert-type-classifier/checkpoint-3296",
|
4 |
+
"epoch": 2.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 3296,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.06067961165048544,
|
13 |
+
"grad_norm": 7.083491325378418,
|
14 |
+
"learning_rate": 1.959546925566343e-05,
|
15 |
+
"loss": 1.3291,
|
16 |
+
"step": 100
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.12135922330097088,
|
20 |
+
"grad_norm": 5.435772895812988,
|
21 |
+
"learning_rate": 1.919093851132686e-05,
|
22 |
+
"loss": 0.7948,
|
23 |
+
"step": 200
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.1820388349514563,
|
27 |
+
"grad_norm": 5.531534671783447,
|
28 |
+
"learning_rate": 1.8786407766990295e-05,
|
29 |
+
"loss": 0.59,
|
30 |
+
"step": 300
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.24271844660194175,
|
34 |
+
"grad_norm": 5.0550031661987305,
|
35 |
+
"learning_rate": 1.838187702265372e-05,
|
36 |
+
"loss": 0.5483,
|
37 |
+
"step": 400
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.30339805825242716,
|
41 |
+
"grad_norm": 9.326889038085938,
|
42 |
+
"learning_rate": 1.7977346278317155e-05,
|
43 |
+
"loss": 0.472,
|
44 |
+
"step": 500
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.3640776699029126,
|
48 |
+
"grad_norm": 11.223947525024414,
|
49 |
+
"learning_rate": 1.7572815533980585e-05,
|
50 |
+
"loss": 0.4334,
|
51 |
+
"step": 600
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.42475728155339804,
|
55 |
+
"grad_norm": 6.310647487640381,
|
56 |
+
"learning_rate": 1.7168284789644015e-05,
|
57 |
+
"loss": 0.3993,
|
58 |
+
"step": 700
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.4854368932038835,
|
62 |
+
"grad_norm": 13.495203018188477,
|
63 |
+
"learning_rate": 1.6763754045307445e-05,
|
64 |
+
"loss": 0.3976,
|
65 |
+
"step": 800
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.5461165048543689,
|
69 |
+
"grad_norm": 9.985793113708496,
|
70 |
+
"learning_rate": 1.6359223300970874e-05,
|
71 |
+
"loss": 0.4045,
|
72 |
+
"step": 900
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.6067961165048543,
|
76 |
+
"grad_norm": 5.920831203460693,
|
77 |
+
"learning_rate": 1.5954692556634304e-05,
|
78 |
+
"loss": 0.3661,
|
79 |
+
"step": 1000
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.6674757281553398,
|
83 |
+
"grad_norm": 7.346311092376709,
|
84 |
+
"learning_rate": 1.5550161812297734e-05,
|
85 |
+
"loss": 0.3527,
|
86 |
+
"step": 1100
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.7281553398058253,
|
90 |
+
"grad_norm": 7.223079681396484,
|
91 |
+
"learning_rate": 1.5145631067961166e-05,
|
92 |
+
"loss": 0.361,
|
93 |
+
"step": 1200
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.7888349514563107,
|
97 |
+
"grad_norm": 5.1326799392700195,
|
98 |
+
"learning_rate": 1.4741100323624598e-05,
|
99 |
+
"loss": 0.3892,
|
100 |
+
"step": 1300
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.8495145631067961,
|
104 |
+
"grad_norm": 7.782454967498779,
|
105 |
+
"learning_rate": 1.4336569579288026e-05,
|
106 |
+
"loss": 0.3502,
|
107 |
+
"step": 1400
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.9101941747572816,
|
111 |
+
"grad_norm": 8.951793670654297,
|
112 |
+
"learning_rate": 1.3932038834951458e-05,
|
113 |
+
"loss": 0.2934,
|
114 |
+
"step": 1500
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.970873786407767,
|
118 |
+
"grad_norm": 9.385202407836914,
|
119 |
+
"learning_rate": 1.3527508090614887e-05,
|
120 |
+
"loss": 0.3436,
|
121 |
+
"step": 1600
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 1.0,
|
125 |
+
"eval_accuracy": 0.8803116147308782,
|
126 |
+
"eval_f1": 0.87963672068735,
|
127 |
+
"eval_loss": 0.30396342277526855,
|
128 |
+
"eval_precision": 0.8827457335389428,
|
129 |
+
"eval_recall": 0.8803116147308782,
|
130 |
+
"eval_runtime": 21.997,
|
131 |
+
"eval_samples_per_second": 256.762,
|
132 |
+
"eval_steps_per_second": 16.048,
|
133 |
+
"step": 1648
|
134 |
+
},
|
135 |
+
{
|
136 |
+
"epoch": 1.0315533980582525,
|
137 |
+
"grad_norm": 11.795103073120117,
|
138 |
+
"learning_rate": 1.3122977346278319e-05,
|
139 |
+
"loss": 0.2579,
|
140 |
+
"step": 1700
|
141 |
+
},
|
142 |
+
{
|
143 |
+
"epoch": 1.0922330097087378,
|
144 |
+
"grad_norm": 3.647850751876831,
|
145 |
+
"learning_rate": 1.2718446601941749e-05,
|
146 |
+
"loss": 0.2231,
|
147 |
+
"step": 1800
|
148 |
+
},
|
149 |
+
{
|
150 |
+
"epoch": 1.1529126213592233,
|
151 |
+
"grad_norm": 6.818081855773926,
|
152 |
+
"learning_rate": 1.2313915857605179e-05,
|
153 |
+
"loss": 0.2381,
|
154 |
+
"step": 1900
|
155 |
+
},
|
156 |
+
{
|
157 |
+
"epoch": 1.2135922330097086,
|
158 |
+
"grad_norm": 25.645702362060547,
|
159 |
+
"learning_rate": 1.190938511326861e-05,
|
160 |
+
"loss": 0.2778,
|
161 |
+
"step": 2000
|
162 |
+
},
|
163 |
+
{
|
164 |
+
"epoch": 1.2742718446601942,
|
165 |
+
"grad_norm": 20.234249114990234,
|
166 |
+
"learning_rate": 1.1504854368932039e-05,
|
167 |
+
"loss": 0.2127,
|
168 |
+
"step": 2100
|
169 |
+
},
|
170 |
+
{
|
171 |
+
"epoch": 1.3349514563106797,
|
172 |
+
"grad_norm": 5.801571846008301,
|
173 |
+
"learning_rate": 1.110032362459547e-05,
|
174 |
+
"loss": 0.2406,
|
175 |
+
"step": 2200
|
176 |
+
},
|
177 |
+
{
|
178 |
+
"epoch": 1.395631067961165,
|
179 |
+
"grad_norm": 3.5099356174468994,
|
180 |
+
"learning_rate": 1.0695792880258899e-05,
|
181 |
+
"loss": 0.2109,
|
182 |
+
"step": 2300
|
183 |
+
},
|
184 |
+
{
|
185 |
+
"epoch": 1.4563106796116505,
|
186 |
+
"grad_norm": 13.334369659423828,
|
187 |
+
"learning_rate": 1.029126213592233e-05,
|
188 |
+
"loss": 0.2268,
|
189 |
+
"step": 2400
|
190 |
+
},
|
191 |
+
{
|
192 |
+
"epoch": 1.516990291262136,
|
193 |
+
"grad_norm": 1.7576143741607666,
|
194 |
+
"learning_rate": 9.88673139158576e-06,
|
195 |
+
"loss": 0.2549,
|
196 |
+
"step": 2500
|
197 |
+
},
|
198 |
+
{
|
199 |
+
"epoch": 1.5776699029126213,
|
200 |
+
"grad_norm": 6.285398006439209,
|
201 |
+
"learning_rate": 9.482200647249192e-06,
|
202 |
+
"loss": 0.2078,
|
203 |
+
"step": 2600
|
204 |
+
},
|
205 |
+
{
|
206 |
+
"epoch": 1.6383495145631068,
|
207 |
+
"grad_norm": 10.36067008972168,
|
208 |
+
"learning_rate": 9.077669902912622e-06,
|
209 |
+
"loss": 0.2391,
|
210 |
+
"step": 2700
|
211 |
+
},
|
212 |
+
{
|
213 |
+
"epoch": 1.6990291262135924,
|
214 |
+
"grad_norm": 7.475758075714111,
|
215 |
+
"learning_rate": 8.673139158576054e-06,
|
216 |
+
"loss": 0.2334,
|
217 |
+
"step": 2800
|
218 |
+
},
|
219 |
+
{
|
220 |
+
"epoch": 1.7597087378640777,
|
221 |
+
"grad_norm": 16.30022430419922,
|
222 |
+
"learning_rate": 8.268608414239483e-06,
|
223 |
+
"loss": 0.2316,
|
224 |
+
"step": 2900
|
225 |
+
},
|
226 |
+
{
|
227 |
+
"epoch": 1.820388349514563,
|
228 |
+
"grad_norm": 4.486410617828369,
|
229 |
+
"learning_rate": 7.864077669902913e-06,
|
230 |
+
"loss": 0.2239,
|
231 |
+
"step": 3000
|
232 |
+
},
|
233 |
+
{
|
234 |
+
"epoch": 1.8810679611650487,
|
235 |
+
"grad_norm": 24.858966827392578,
|
236 |
+
"learning_rate": 7.459546925566343e-06,
|
237 |
+
"loss": 0.2162,
|
238 |
+
"step": 3100
|
239 |
+
},
|
240 |
+
{
|
241 |
+
"epoch": 1.941747572815534,
|
242 |
+
"grad_norm": 18.323312759399414,
|
243 |
+
"learning_rate": 7.055016181229773e-06,
|
244 |
+
"loss": 0.2452,
|
245 |
+
"step": 3200
|
246 |
+
},
|
247 |
+
{
|
248 |
+
"epoch": 2.0,
|
249 |
+
"eval_accuracy": 0.9008498583569405,
|
250 |
+
"eval_f1": 0.8996119975479232,
|
251 |
+
"eval_loss": 0.2864818572998047,
|
252 |
+
"eval_precision": 0.9011561063745416,
|
253 |
+
"eval_recall": 0.9008498583569405,
|
254 |
+
"eval_runtime": 22.0341,
|
255 |
+
"eval_samples_per_second": 256.33,
|
256 |
+
"eval_steps_per_second": 16.021,
|
257 |
+
"step": 3296
|
258 |
+
}
|
259 |
+
],
|
260 |
+
"logging_steps": 100,
|
261 |
+
"max_steps": 4944,
|
262 |
+
"num_input_tokens_seen": 0,
|
263 |
+
"num_train_epochs": 3,
|
264 |
+
"save_steps": 500,
|
265 |
+
"stateful_callbacks": {
|
266 |
+
"EarlyStoppingCallback": {
|
267 |
+
"args": {
|
268 |
+
"early_stopping_patience": 2,
|
269 |
+
"early_stopping_threshold": 0.0
|
270 |
+
},
|
271 |
+
"attributes": {
|
272 |
+
"early_stopping_patience_counter": 0
|
273 |
+
}
|
274 |
+
},
|
275 |
+
"TrainerControl": {
|
276 |
+
"args": {
|
277 |
+
"should_epoch_stop": false,
|
278 |
+
"should_evaluate": false,
|
279 |
+
"should_log": false,
|
280 |
+
"should_save": true,
|
281 |
+
"should_training_stop": false
|
282 |
+
},
|
283 |
+
"attributes": {}
|
284 |
+
}
|
285 |
+
},
|
286 |
+
"total_flos": 1745811035916288.0,
|
287 |
+
"train_batch_size": 16,
|
288 |
+
"trial_name": null,
|
289 |
+
"trial_params": null
|
290 |
+
}
|
checkpoint-3296/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:609d9e093ff10cb72ff2ceab0b60b959ffb5a8e69641d5f5afc035f09f45f51b
|
3 |
+
size 5304
|
checkpoint-4944/config.json
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "distilbert-base-multilingual-cased",
|
3 |
+
"activation": "gelu",
|
4 |
+
"architectures": [
|
5 |
+
"DistilBertForSequenceClassification"
|
6 |
+
],
|
7 |
+
"attention_dropout": 0.1,
|
8 |
+
"dim": 768,
|
9 |
+
"dropout": 0.1,
|
10 |
+
"hidden_dim": 3072,
|
11 |
+
"id2label": {
|
12 |
+
"0": "LABEL_0",
|
13 |
+
"1": "LABEL_1",
|
14 |
+
"2": "LABEL_2",
|
15 |
+
"3": "LABEL_3",
|
16 |
+
"4": "LABEL_4",
|
17 |
+
"5": "LABEL_5",
|
18 |
+
"6": "LABEL_6"
|
19 |
+
},
|
20 |
+
"initializer_range": 0.02,
|
21 |
+
"label2id": {
|
22 |
+
"LABEL_0": 0,
|
23 |
+
"LABEL_1": 1,
|
24 |
+
"LABEL_2": 2,
|
25 |
+
"LABEL_3": 3,
|
26 |
+
"LABEL_4": 4,
|
27 |
+
"LABEL_5": 5,
|
28 |
+
"LABEL_6": 6
|
29 |
+
},
|
30 |
+
"max_position_embeddings": 512,
|
31 |
+
"model_type": "distilbert",
|
32 |
+
"n_heads": 12,
|
33 |
+
"n_layers": 6,
|
34 |
+
"output_past": true,
|
35 |
+
"pad_token_id": 0,
|
36 |
+
"problem_type": "single_label_classification",
|
37 |
+
"qa_dropout": 0.1,
|
38 |
+
"seq_classif_dropout": 0.2,
|
39 |
+
"sinusoidal_pos_embds": false,
|
40 |
+
"tie_weights_": true,
|
41 |
+
"torch_dtype": "float32",
|
42 |
+
"transformers_version": "4.49.0",
|
43 |
+
"vocab_size": 119547
|
44 |
+
}
|
checkpoint-4944/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6c5eb116208f77aeac8c1dd22271929d9fdbc47b8f91e8b401b6e8af3a4d4349
|
3 |
+
size 541332756
|
checkpoint-4944/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8155baa2fbbdc05e6df9c79413c0b6e2ea203210035c4355d85c0b2bf4e99e22
|
3 |
+
size 1082727610
|
checkpoint-4944/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:07a5237c61fd9606d1962942985d8ec394dba78a7c8ffa761c2b48b2aa04b2f5
|
3 |
+
size 14244
|
checkpoint-4944/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ca493678282795f2e99cbd4840bd077221fa7cf771f4e64768bf266ad9336fef
|
3 |
+
size 1064
|
checkpoint-4944/trainer_state.json
ADDED
@@ -0,0 +1,421 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": 0.905103252820856,
|
3 |
+
"best_model_checkpoint": "distilbert-type-classifier/checkpoint-4944",
|
4 |
+
"epoch": 3.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 4944,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.06067961165048544,
|
13 |
+
"grad_norm": 7.083491325378418,
|
14 |
+
"learning_rate": 1.959546925566343e-05,
|
15 |
+
"loss": 1.3291,
|
16 |
+
"step": 100
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.12135922330097088,
|
20 |
+
"grad_norm": 5.435772895812988,
|
21 |
+
"learning_rate": 1.919093851132686e-05,
|
22 |
+
"loss": 0.7948,
|
23 |
+
"step": 200
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.1820388349514563,
|
27 |
+
"grad_norm": 5.531534671783447,
|
28 |
+
"learning_rate": 1.8786407766990295e-05,
|
29 |
+
"loss": 0.59,
|
30 |
+
"step": 300
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.24271844660194175,
|
34 |
+
"grad_norm": 5.0550031661987305,
|
35 |
+
"learning_rate": 1.838187702265372e-05,
|
36 |
+
"loss": 0.5483,
|
37 |
+
"step": 400
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.30339805825242716,
|
41 |
+
"grad_norm": 9.326889038085938,
|
42 |
+
"learning_rate": 1.7977346278317155e-05,
|
43 |
+
"loss": 0.472,
|
44 |
+
"step": 500
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.3640776699029126,
|
48 |
+
"grad_norm": 11.223947525024414,
|
49 |
+
"learning_rate": 1.7572815533980585e-05,
|
50 |
+
"loss": 0.4334,
|
51 |
+
"step": 600
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.42475728155339804,
|
55 |
+
"grad_norm": 6.310647487640381,
|
56 |
+
"learning_rate": 1.7168284789644015e-05,
|
57 |
+
"loss": 0.3993,
|
58 |
+
"step": 700
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.4854368932038835,
|
62 |
+
"grad_norm": 13.495203018188477,
|
63 |
+
"learning_rate": 1.6763754045307445e-05,
|
64 |
+
"loss": 0.3976,
|
65 |
+
"step": 800
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.5461165048543689,
|
69 |
+
"grad_norm": 9.985793113708496,
|
70 |
+
"learning_rate": 1.6359223300970874e-05,
|
71 |
+
"loss": 0.4045,
|
72 |
+
"step": 900
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.6067961165048543,
|
76 |
+
"grad_norm": 5.920831203460693,
|
77 |
+
"learning_rate": 1.5954692556634304e-05,
|
78 |
+
"loss": 0.3661,
|
79 |
+
"step": 1000
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.6674757281553398,
|
83 |
+
"grad_norm": 7.346311092376709,
|
84 |
+
"learning_rate": 1.5550161812297734e-05,
|
85 |
+
"loss": 0.3527,
|
86 |
+
"step": 1100
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.7281553398058253,
|
90 |
+
"grad_norm": 7.223079681396484,
|
91 |
+
"learning_rate": 1.5145631067961166e-05,
|
92 |
+
"loss": 0.361,
|
93 |
+
"step": 1200
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.7888349514563107,
|
97 |
+
"grad_norm": 5.1326799392700195,
|
98 |
+
"learning_rate": 1.4741100323624598e-05,
|
99 |
+
"loss": 0.3892,
|
100 |
+
"step": 1300
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.8495145631067961,
|
104 |
+
"grad_norm": 7.782454967498779,
|
105 |
+
"learning_rate": 1.4336569579288026e-05,
|
106 |
+
"loss": 0.3502,
|
107 |
+
"step": 1400
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.9101941747572816,
|
111 |
+
"grad_norm": 8.951793670654297,
|
112 |
+
"learning_rate": 1.3932038834951458e-05,
|
113 |
+
"loss": 0.2934,
|
114 |
+
"step": 1500
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.970873786407767,
|
118 |
+
"grad_norm": 9.385202407836914,
|
119 |
+
"learning_rate": 1.3527508090614887e-05,
|
120 |
+
"loss": 0.3436,
|
121 |
+
"step": 1600
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 1.0,
|
125 |
+
"eval_accuracy": 0.8803116147308782,
|
126 |
+
"eval_f1": 0.87963672068735,
|
127 |
+
"eval_loss": 0.30396342277526855,
|
128 |
+
"eval_precision": 0.8827457335389428,
|
129 |
+
"eval_recall": 0.8803116147308782,
|
130 |
+
"eval_runtime": 21.997,
|
131 |
+
"eval_samples_per_second": 256.762,
|
132 |
+
"eval_steps_per_second": 16.048,
|
133 |
+
"step": 1648
|
134 |
+
},
|
135 |
+
{
|
136 |
+
"epoch": 1.0315533980582525,
|
137 |
+
"grad_norm": 11.795103073120117,
|
138 |
+
"learning_rate": 1.3122977346278319e-05,
|
139 |
+
"loss": 0.2579,
|
140 |
+
"step": 1700
|
141 |
+
},
|
142 |
+
{
|
143 |
+
"epoch": 1.0922330097087378,
|
144 |
+
"grad_norm": 3.647850751876831,
|
145 |
+
"learning_rate": 1.2718446601941749e-05,
|
146 |
+
"loss": 0.2231,
|
147 |
+
"step": 1800
|
148 |
+
},
|
149 |
+
{
|
150 |
+
"epoch": 1.1529126213592233,
|
151 |
+
"grad_norm": 6.818081855773926,
|
152 |
+
"learning_rate": 1.2313915857605179e-05,
|
153 |
+
"loss": 0.2381,
|
154 |
+
"step": 1900
|
155 |
+
},
|
156 |
+
{
|
157 |
+
"epoch": 1.2135922330097086,
|
158 |
+
"grad_norm": 25.645702362060547,
|
159 |
+
"learning_rate": 1.190938511326861e-05,
|
160 |
+
"loss": 0.2778,
|
161 |
+
"step": 2000
|
162 |
+
},
|
163 |
+
{
|
164 |
+
"epoch": 1.2742718446601942,
|
165 |
+
"grad_norm": 20.234249114990234,
|
166 |
+
"learning_rate": 1.1504854368932039e-05,
|
167 |
+
"loss": 0.2127,
|
168 |
+
"step": 2100
|
169 |
+
},
|
170 |
+
{
|
171 |
+
"epoch": 1.3349514563106797,
|
172 |
+
"grad_norm": 5.801571846008301,
|
173 |
+
"learning_rate": 1.110032362459547e-05,
|
174 |
+
"loss": 0.2406,
|
175 |
+
"step": 2200
|
176 |
+
},
|
177 |
+
{
|
178 |
+
"epoch": 1.395631067961165,
|
179 |
+
"grad_norm": 3.5099356174468994,
|
180 |
+
"learning_rate": 1.0695792880258899e-05,
|
181 |
+
"loss": 0.2109,
|
182 |
+
"step": 2300
|
183 |
+
},
|
184 |
+
{
|
185 |
+
"epoch": 1.4563106796116505,
|
186 |
+
"grad_norm": 13.334369659423828,
|
187 |
+
"learning_rate": 1.029126213592233e-05,
|
188 |
+
"loss": 0.2268,
|
189 |
+
"step": 2400
|
190 |
+
},
|
191 |
+
{
|
192 |
+
"epoch": 1.516990291262136,
|
193 |
+
"grad_norm": 1.7576143741607666,
|
194 |
+
"learning_rate": 9.88673139158576e-06,
|
195 |
+
"loss": 0.2549,
|
196 |
+
"step": 2500
|
197 |
+
},
|
198 |
+
{
|
199 |
+
"epoch": 1.5776699029126213,
|
200 |
+
"grad_norm": 6.285398006439209,
|
201 |
+
"learning_rate": 9.482200647249192e-06,
|
202 |
+
"loss": 0.2078,
|
203 |
+
"step": 2600
|
204 |
+
},
|
205 |
+
{
|
206 |
+
"epoch": 1.6383495145631068,
|
207 |
+
"grad_norm": 10.36067008972168,
|
208 |
+
"learning_rate": 9.077669902912622e-06,
|
209 |
+
"loss": 0.2391,
|
210 |
+
"step": 2700
|
211 |
+
},
|
212 |
+
{
|
213 |
+
"epoch": 1.6990291262135924,
|
214 |
+
"grad_norm": 7.475758075714111,
|
215 |
+
"learning_rate": 8.673139158576054e-06,
|
216 |
+
"loss": 0.2334,
|
217 |
+
"step": 2800
|
218 |
+
},
|
219 |
+
{
|
220 |
+
"epoch": 1.7597087378640777,
|
221 |
+
"grad_norm": 16.30022430419922,
|
222 |
+
"learning_rate": 8.268608414239483e-06,
|
223 |
+
"loss": 0.2316,
|
224 |
+
"step": 2900
|
225 |
+
},
|
226 |
+
{
|
227 |
+
"epoch": 1.820388349514563,
|
228 |
+
"grad_norm": 4.486410617828369,
|
229 |
+
"learning_rate": 7.864077669902913e-06,
|
230 |
+
"loss": 0.2239,
|
231 |
+
"step": 3000
|
232 |
+
},
|
233 |
+
{
|
234 |
+
"epoch": 1.8810679611650487,
|
235 |
+
"grad_norm": 24.858966827392578,
|
236 |
+
"learning_rate": 7.459546925566343e-06,
|
237 |
+
"loss": 0.2162,
|
238 |
+
"step": 3100
|
239 |
+
},
|
240 |
+
{
|
241 |
+
"epoch": 1.941747572815534,
|
242 |
+
"grad_norm": 18.323312759399414,
|
243 |
+
"learning_rate": 7.055016181229773e-06,
|
244 |
+
"loss": 0.2452,
|
245 |
+
"step": 3200
|
246 |
+
},
|
247 |
+
{
|
248 |
+
"epoch": 2.0,
|
249 |
+
"eval_accuracy": 0.9008498583569405,
|
250 |
+
"eval_f1": 0.8996119975479232,
|
251 |
+
"eval_loss": 0.2864818572998047,
|
252 |
+
"eval_precision": 0.9011561063745416,
|
253 |
+
"eval_recall": 0.9008498583569405,
|
254 |
+
"eval_runtime": 22.0341,
|
255 |
+
"eval_samples_per_second": 256.33,
|
256 |
+
"eval_steps_per_second": 16.021,
|
257 |
+
"step": 3296
|
258 |
+
},
|
259 |
+
{
|
260 |
+
"epoch": 2.0024271844660193,
|
261 |
+
"grad_norm": 22.042865753173828,
|
262 |
+
"learning_rate": 6.650485436893205e-06,
|
263 |
+
"loss": 0.1939,
|
264 |
+
"step": 3300
|
265 |
+
},
|
266 |
+
{
|
267 |
+
"epoch": 2.063106796116505,
|
268 |
+
"grad_norm": 3.2583961486816406,
|
269 |
+
"learning_rate": 6.245954692556635e-06,
|
270 |
+
"loss": 0.1526,
|
271 |
+
"step": 3400
|
272 |
+
},
|
273 |
+
{
|
274 |
+
"epoch": 2.1237864077669903,
|
275 |
+
"grad_norm": 16.607648849487305,
|
276 |
+
"learning_rate": 5.841423948220065e-06,
|
277 |
+
"loss": 0.1431,
|
278 |
+
"step": 3500
|
279 |
+
},
|
280 |
+
{
|
281 |
+
"epoch": 2.1844660194174756,
|
282 |
+
"grad_norm": 5.757761478424072,
|
283 |
+
"learning_rate": 5.436893203883496e-06,
|
284 |
+
"loss": 0.1566,
|
285 |
+
"step": 3600
|
286 |
+
},
|
287 |
+
{
|
288 |
+
"epoch": 2.2451456310679614,
|
289 |
+
"grad_norm": 10.476051330566406,
|
290 |
+
"learning_rate": 5.0323624595469255e-06,
|
291 |
+
"loss": 0.1645,
|
292 |
+
"step": 3700
|
293 |
+
},
|
294 |
+
{
|
295 |
+
"epoch": 2.3058252427184467,
|
296 |
+
"grad_norm": 3.0033633708953857,
|
297 |
+
"learning_rate": 4.627831715210356e-06,
|
298 |
+
"loss": 0.1461,
|
299 |
+
"step": 3800
|
300 |
+
},
|
301 |
+
{
|
302 |
+
"epoch": 2.366504854368932,
|
303 |
+
"grad_norm": 18.662181854248047,
|
304 |
+
"learning_rate": 4.223300970873786e-06,
|
305 |
+
"loss": 0.1552,
|
306 |
+
"step": 3900
|
307 |
+
},
|
308 |
+
{
|
309 |
+
"epoch": 2.4271844660194173,
|
310 |
+
"grad_norm": 3.020024538040161,
|
311 |
+
"learning_rate": 3.818770226537217e-06,
|
312 |
+
"loss": 0.1793,
|
313 |
+
"step": 4000
|
314 |
+
},
|
315 |
+
{
|
316 |
+
"epoch": 2.487864077669903,
|
317 |
+
"grad_norm": 4.562654495239258,
|
318 |
+
"learning_rate": 3.4142394822006474e-06,
|
319 |
+
"loss": 0.143,
|
320 |
+
"step": 4100
|
321 |
+
},
|
322 |
+
{
|
323 |
+
"epoch": 2.5485436893203883,
|
324 |
+
"grad_norm": 11.560980796813965,
|
325 |
+
"learning_rate": 3.0097087378640778e-06,
|
326 |
+
"loss": 0.1584,
|
327 |
+
"step": 4200
|
328 |
+
},
|
329 |
+
{
|
330 |
+
"epoch": 2.6092233009708736,
|
331 |
+
"grad_norm": 6.995797634124756,
|
332 |
+
"learning_rate": 2.6051779935275086e-06,
|
333 |
+
"loss": 0.1488,
|
334 |
+
"step": 4300
|
335 |
+
},
|
336 |
+
{
|
337 |
+
"epoch": 2.6699029126213594,
|
338 |
+
"grad_norm": 2.360529661178589,
|
339 |
+
"learning_rate": 2.2006472491909385e-06,
|
340 |
+
"loss": 0.1338,
|
341 |
+
"step": 4400
|
342 |
+
},
|
343 |
+
{
|
344 |
+
"epoch": 2.7305825242718447,
|
345 |
+
"grad_norm": 19.43924903869629,
|
346 |
+
"learning_rate": 1.796116504854369e-06,
|
347 |
+
"loss": 0.1531,
|
348 |
+
"step": 4500
|
349 |
+
},
|
350 |
+
{
|
351 |
+
"epoch": 2.79126213592233,
|
352 |
+
"grad_norm": 6.97533655166626,
|
353 |
+
"learning_rate": 1.3915857605177997e-06,
|
354 |
+
"loss": 0.1586,
|
355 |
+
"step": 4600
|
356 |
+
},
|
357 |
+
{
|
358 |
+
"epoch": 2.8519417475728153,
|
359 |
+
"grad_norm": 1.0474483966827393,
|
360 |
+
"learning_rate": 9.870550161812298e-07,
|
361 |
+
"loss": 0.1455,
|
362 |
+
"step": 4700
|
363 |
+
},
|
364 |
+
{
|
365 |
+
"epoch": 2.912621359223301,
|
366 |
+
"grad_norm": 25.780860900878906,
|
367 |
+
"learning_rate": 5.825242718446603e-07,
|
368 |
+
"loss": 0.1604,
|
369 |
+
"step": 4800
|
370 |
+
},
|
371 |
+
{
|
372 |
+
"epoch": 2.9733009708737863,
|
373 |
+
"grad_norm": 5.437297821044922,
|
374 |
+
"learning_rate": 1.7799352750809063e-07,
|
375 |
+
"loss": 0.1497,
|
376 |
+
"step": 4900
|
377 |
+
},
|
378 |
+
{
|
379 |
+
"epoch": 3.0,
|
380 |
+
"eval_accuracy": 0.9061614730878187,
|
381 |
+
"eval_f1": 0.905103252820856,
|
382 |
+
"eval_loss": 0.313060998916626,
|
383 |
+
"eval_precision": 0.9060485063772568,
|
384 |
+
"eval_recall": 0.9061614730878187,
|
385 |
+
"eval_runtime": 23.35,
|
386 |
+
"eval_samples_per_second": 241.884,
|
387 |
+
"eval_steps_per_second": 15.118,
|
388 |
+
"step": 4944
|
389 |
+
}
|
390 |
+
],
|
391 |
+
"logging_steps": 100,
|
392 |
+
"max_steps": 4944,
|
393 |
+
"num_input_tokens_seen": 0,
|
394 |
+
"num_train_epochs": 3,
|
395 |
+
"save_steps": 500,
|
396 |
+
"stateful_callbacks": {
|
397 |
+
"EarlyStoppingCallback": {
|
398 |
+
"args": {
|
399 |
+
"early_stopping_patience": 2,
|
400 |
+
"early_stopping_threshold": 0.0
|
401 |
+
},
|
402 |
+
"attributes": {
|
403 |
+
"early_stopping_patience_counter": 0
|
404 |
+
}
|
405 |
+
},
|
406 |
+
"TrainerControl": {
|
407 |
+
"args": {
|
408 |
+
"should_epoch_stop": false,
|
409 |
+
"should_evaluate": false,
|
410 |
+
"should_log": false,
|
411 |
+
"should_save": true,
|
412 |
+
"should_training_stop": true
|
413 |
+
},
|
414 |
+
"attributes": {}
|
415 |
+
}
|
416 |
+
},
|
417 |
+
"total_flos": 2618716553874432.0,
|
418 |
+
"train_batch_size": 16,
|
419 |
+
"trial_name": null,
|
420 |
+
"trial_params": null
|
421 |
+
}
|
checkpoint-4944/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:609d9e093ff10cb72ff2ceab0b60b959ffb5a8e69641d5f5afc035f09f45f51b
|
3 |
+
size 5304
|
config.json
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "distilbert-base-multilingual-cased",
|
3 |
+
"activation": "gelu",
|
4 |
+
"architectures": [
|
5 |
+
"DistilBertForSequenceClassification"
|
6 |
+
],
|
7 |
+
"attention_dropout": 0.1,
|
8 |
+
"dim": 768,
|
9 |
+
"dropout": 0.1,
|
10 |
+
"hidden_dim": 3072,
|
11 |
+
"id2label": {
|
12 |
+
"0": "LABEL_0",
|
13 |
+
"1": "LABEL_1",
|
14 |
+
"2": "LABEL_2",
|
15 |
+
"3": "LABEL_3",
|
16 |
+
"4": "LABEL_4",
|
17 |
+
"5": "LABEL_5",
|
18 |
+
"6": "LABEL_6"
|
19 |
+
},
|
20 |
+
"initializer_range": 0.02,
|
21 |
+
"label2id": {
|
22 |
+
"LABEL_0": 0,
|
23 |
+
"LABEL_1": 1,
|
24 |
+
"LABEL_2": 2,
|
25 |
+
"LABEL_3": 3,
|
26 |
+
"LABEL_4": 4,
|
27 |
+
"LABEL_5": 5,
|
28 |
+
"LABEL_6": 6
|
29 |
+
},
|
30 |
+
"max_position_embeddings": 512,
|
31 |
+
"model_type": "distilbert",
|
32 |
+
"n_heads": 12,
|
33 |
+
"n_layers": 6,
|
34 |
+
"output_past": true,
|
35 |
+
"pad_token_id": 0,
|
36 |
+
"problem_type": "single_label_classification",
|
37 |
+
"qa_dropout": 0.1,
|
38 |
+
"seq_classif_dropout": 0.2,
|
39 |
+
"sinusoidal_pos_embds": false,
|
40 |
+
"tie_weights_": true,
|
41 |
+
"torch_dtype": "float32",
|
42 |
+
"transformers_version": "4.49.0",
|
43 |
+
"vocab_size": 119547
|
44 |
+
}
|
confusion_matrix.png
ADDED
![]() |
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6c5eb116208f77aeac8c1dd22271929d9fdbc47b8f91e8b401b6e8af3a4d4349
|
3 |
+
size 541332756
|
special_tokens_map.json
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"cls_token": "[CLS]",
|
3 |
+
"mask_token": "[MASK]",
|
4 |
+
"pad_token": "[PAD]",
|
5 |
+
"sep_token": "[SEP]",
|
6 |
+
"unk_token": "[UNK]"
|
7 |
+
}
|
tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
tokenizer_config.json
ADDED
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"added_tokens_decoder": {
|
3 |
+
"0": {
|
4 |
+
"content": "[PAD]",
|
5 |
+
"lstrip": false,
|
6 |
+
"normalized": false,
|
7 |
+
"rstrip": false,
|
8 |
+
"single_word": false,
|
9 |
+
"special": true
|
10 |
+
},
|
11 |
+
"100": {
|
12 |
+
"content": "[UNK]",
|
13 |
+
"lstrip": false,
|
14 |
+
"normalized": false,
|
15 |
+
"rstrip": false,
|
16 |
+
"single_word": false,
|
17 |
+
"special": true
|
18 |
+
},
|
19 |
+
"101": {
|
20 |
+
"content": "[CLS]",
|
21 |
+
"lstrip": false,
|
22 |
+
"normalized": false,
|
23 |
+
"rstrip": false,
|
24 |
+
"single_word": false,
|
25 |
+
"special": true
|
26 |
+
},
|
27 |
+
"102": {
|
28 |
+
"content": "[SEP]",
|
29 |
+
"lstrip": false,
|
30 |
+
"normalized": false,
|
31 |
+
"rstrip": false,
|
32 |
+
"single_word": false,
|
33 |
+
"special": true
|
34 |
+
},
|
35 |
+
"103": {
|
36 |
+
"content": "[MASK]",
|
37 |
+
"lstrip": false,
|
38 |
+
"normalized": false,
|
39 |
+
"rstrip": false,
|
40 |
+
"single_word": false,
|
41 |
+
"special": true
|
42 |
+
}
|
43 |
+
},
|
44 |
+
"clean_up_tokenization_spaces": false,
|
45 |
+
"cls_token": "[CLS]",
|
46 |
+
"do_lower_case": false,
|
47 |
+
"extra_special_tokens": {},
|
48 |
+
"mask_token": "[MASK]",
|
49 |
+
"model_max_length": 512,
|
50 |
+
"pad_token": "[PAD]",
|
51 |
+
"sep_token": "[SEP]",
|
52 |
+
"strip_accents": null,
|
53 |
+
"tokenize_chinese_chars": true,
|
54 |
+
"tokenizer_class": "DistilBertTokenizer",
|
55 |
+
"unk_token": "[UNK]"
|
56 |
+
}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:609d9e093ff10cb72ff2ceab0b60b959ffb5a8e69641d5f5afc035f09f45f51b
|
3 |
+
size 5304
|
type_mapping.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"0": "coherent", "1": "grammatical_errors", "2": "random_bytes", "3": "random_tokens", "4": "random_words", "5": "run_on", "6": "word_soup"}
|
vocab.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|