Zekun Wu commited on
Commit
2d9bac1
1 Parent(s): 47ce046
config.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilbert-base-uncased",
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
+ "7": "LABEL_7",
20
+ "8": "LABEL_8"
21
+ },
22
+ "initializer_range": 0.02,
23
+ "label2id": {
24
+ "LABEL_0": 0,
25
+ "LABEL_1": 1,
26
+ "LABEL_2": 2,
27
+ "LABEL_3": 3,
28
+ "LABEL_4": 4,
29
+ "LABEL_5": 5,
30
+ "LABEL_6": 6,
31
+ "LABEL_7": 7,
32
+ "LABEL_8": 8
33
+ },
34
+ "max_position_embeddings": 512,
35
+ "model_type": "distilbert",
36
+ "n_heads": 12,
37
+ "n_layers": 6,
38
+ "pad_token_id": 0,
39
+ "problem_type": "single_label_classification",
40
+ "qa_dropout": 0.1,
41
+ "seq_classif_dropout": 0.2,
42
+ "sinusoidal_pos_embds": false,
43
+ "tie_weights_": true,
44
+ "torch_dtype": "float32",
45
+ "transformers_version": "4.30.2",
46
+ "vocab_size": 30522
47
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fe6d44ed8eadd7b3fa298e0b516c01635f64e8d63cb962aecfd656af54bc9df
3
+ size 535744069
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2bea91ab55815e992847f99e4cfa09b525f0ae38062ae428c373962f2f6ff77c
3
+ size 267877037
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b30dbcef122e942f8a8df2b8e1c57b3ec194d344225213f40f2c328c6925c6d6
3
+ size 14575
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf7f11b4a721200e4855c318b8001b95008046cba699011100f3bf012c01c546
3
+ size 627
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,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "clean_up_tokenization_spaces": true,
3
+ "cls_token": "[CLS]",
4
+ "do_lower_case": true,
5
+ "mask_token": "[MASK]",
6
+ "model_max_length": 512,
7
+ "pad_token": "[PAD]",
8
+ "sep_token": "[SEP]",
9
+ "strip_accents": null,
10
+ "tokenize_chinese_chars": true,
11
+ "tokenizer_class": "DistilBertTokenizer",
12
+ "unk_token": "[UNK]"
13
+ }
trainer_state.json ADDED
@@ -0,0 +1,100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.49603304266929626,
3
+ "best_model_checkpoint": "MD_SL_best_model/checkpoint-5276",
4
+ "epoch": 2.0,
5
+ "global_step": 5276,
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.19,
12
+ "learning_rate": 1.936820823856457e-05,
13
+ "loss": 1.0828,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 0.38,
18
+ "learning_rate": 1.873641647712914e-05,
19
+ "loss": 0.7086,
20
+ "step": 1000
21
+ },
22
+ {
23
+ "epoch": 0.57,
24
+ "learning_rate": 1.8104624715693708e-05,
25
+ "loss": 0.669,
26
+ "step": 1500
27
+ },
28
+ {
29
+ "epoch": 0.76,
30
+ "learning_rate": 1.747283295425828e-05,
31
+ "loss": 0.6264,
32
+ "step": 2000
33
+ },
34
+ {
35
+ "epoch": 0.95,
36
+ "learning_rate": 1.6841041192822848e-05,
37
+ "loss": 0.5951,
38
+ "step": 2500
39
+ },
40
+ {
41
+ "epoch": 1.0,
42
+ "eval_balanced accuracy": 0.6598598463994036,
43
+ "eval_f1": 0.6477315522294683,
44
+ "eval_loss": 0.5617039799690247,
45
+ "eval_precision": 0.6664522794774816,
46
+ "eval_recall": 0.6598598463994036,
47
+ "eval_runtime": 177.7279,
48
+ "eval_samples_per_second": 59.366,
49
+ "eval_steps_per_second": 3.714,
50
+ "step": 2638
51
+ },
52
+ {
53
+ "epoch": 1.14,
54
+ "learning_rate": 1.6209249431387417e-05,
55
+ "loss": 0.5066,
56
+ "step": 3000
57
+ },
58
+ {
59
+ "epoch": 1.33,
60
+ "learning_rate": 1.5577457669951985e-05,
61
+ "loss": 0.475,
62
+ "step": 3500
63
+ },
64
+ {
65
+ "epoch": 1.52,
66
+ "learning_rate": 1.4945665908516554e-05,
67
+ "loss": 0.4635,
68
+ "step": 4000
69
+ },
70
+ {
71
+ "epoch": 1.71,
72
+ "learning_rate": 1.4313874147081124e-05,
73
+ "loss": 0.4546,
74
+ "step": 4500
75
+ },
76
+ {
77
+ "epoch": 1.9,
78
+ "learning_rate": 1.3682082385645691e-05,
79
+ "loss": 0.4428,
80
+ "step": 5000
81
+ },
82
+ {
83
+ "epoch": 2.0,
84
+ "eval_balanced accuracy": 0.7034186400261718,
85
+ "eval_f1": 0.6980022981890632,
86
+ "eval_loss": 0.49603304266929626,
87
+ "eval_precision": 0.7144313872416078,
88
+ "eval_recall": 0.7034186400261718,
89
+ "eval_runtime": 177.7105,
90
+ "eval_samples_per_second": 59.372,
91
+ "eval_steps_per_second": 3.714,
92
+ "step": 5276
93
+ }
94
+ ],
95
+ "max_steps": 15828,
96
+ "num_train_epochs": 6,
97
+ "total_flos": 1.11816441335808e+16,
98
+ "trial_name": null,
99
+ "trial_params": null
100
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ca839a775bd188814fd1d454dfd76732fcd7c1f4521dc5b5f57416f583db5f0
3
+ size 3963
vocab.txt ADDED
The diff for this file is too large to render. See raw diff