jialicheng commited on
Commit
05561dc
1 Parent(s): d66acce

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: distilbert/distilbert-base-uncased
4
+ tags:
5
+ - generated_from_trainer
6
+ model-index:
7
+ - name: imdb_42
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ # imdb_42
15
+
16
+ This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - eval_loss: 0.0838
19
+ - eval_accuracy: 0.9781
20
+ - eval_runtime: 20.6358
21
+ - eval_samples_per_second: 1211.49
22
+ - eval_steps_per_second: 4.749
23
+ - step: 0
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 5e-05
43
+ - train_batch_size: 32
44
+ - eval_batch_size: 256
45
+ - seed: 42
46
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
+ - lr_scheduler_type: linear
48
+ - num_epochs: 10
49
+
50
+ ### Framework versions
51
+
52
+ - Transformers 4.39.3
53
+ - Pytorch 2.2.2+cu118
54
+ - Datasets 2.18.0
55
+ - Tokenizers 0.15.2
all_results.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "eval_accuracy": 0.97808,
4
+ "eval_loss": 0.0838116779923439,
5
+ "eval_runtime": 20.6358,
6
+ "eval_samples": 25000,
7
+ "eval_samples_per_second": 1211.49,
8
+ "eval_steps_per_second": 4.749,
9
+ "train_loss": 0.07839516550683609,
10
+ "train_runtime": 418.6292,
11
+ "train_samples": 25000,
12
+ "train_samples_per_second": 597.187,
13
+ "train_steps_per_second": 18.68
14
+ }
config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilbert/distilbert-base-uncased",
3
+ "activation": "gelu",
4
+ "architectures": [
5
+ "DistilBertForSequenceClassification"
6
+ ],
7
+ "attention_dropout": 0.1,
8
+ "dim": 768,
9
+ "dropout": 0.1,
10
+ "finetuning_task": "text-classification",
11
+ "hidden_dim": 3072,
12
+ "id2label": {
13
+ "0": "0",
14
+ "1": "1"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "label2id": {
18
+ "0": 0,
19
+ "1": 1
20
+ },
21
+ "max_position_embeddings": 512,
22
+ "model_type": "distilbert",
23
+ "n_heads": 12,
24
+ "n_layers": 6,
25
+ "pad_token_id": 0,
26
+ "problem_type": "single_label_classification",
27
+ "qa_dropout": 0.1,
28
+ "seq_classif_dropout": 0.2,
29
+ "sinusoidal_pos_embds": false,
30
+ "tie_weights_": true,
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.28.0",
33
+ "vocab_size": 30522
34
+ }
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "eval_accuracy": 0.87392,
3
+ "eval_loss": 0.3133154809474945,
4
+ "eval_runtime": 24.0502,
5
+ "eval_samples": 25000,
6
+ "eval_samples_per_second": 1039.49,
7
+ "eval_steps_per_second": 4.075
8
+ }
pred_logit_eval.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa9130026e7c07ecaa7d0570d803efc6848ce04b23096615ce4059c0b8870c3b
3
+ size 200128
pred_logit_test.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa9130026e7c07ecaa7d0570d803efc6848ce04b23096615ce4059c0b8870c3b
3
+ size 200128
pred_logit_train.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c2ee86be5d3df9cd9f69f02973592a6e3f5656a8042a1568de78d8885268a962
3
+ size 200128
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d32ce46471955018916130a430e53ed53ca71df217b10f3db8b5447a15cf5a1d
3
+ size 267854321
runs/Apr21_18-50-36_clu/events.out.tfevents.1713725484.clu ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4777ff8661c69cb27a892e87aba3ada813b05df6621c12b6c0cb81b50845b58
3
+ size 814
runs/Mar18_09-33-46_clu/1710754452.2684553/events.out.tfevents.1710754452.clu ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99fa44f729d1d5f5b88ac783d6a4273a082da823293906f349bf156daf174474
3
+ size 5643
runs/Mar18_09-33-46_clu/events.out.tfevents.1710754452.clu ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93a69d9fea6b888f957bd9cb8f0bed6eb312b31a86e0a1a7fd40a74ace302503
3
+ size 8785
runs/Mar18_09-33-46_clu/events.out.tfevents.1710754879.clu ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58be613696db520d4418e80e443d64022d39cabddc2c2525712b7e6a7edcada2
3
+ size 363
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
+ }
test_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "eval_accuracy": 0.87392,
3
+ "eval_loss": 0.3133154809474945,
4
+ "eval_runtime": 20.5489,
5
+ "eval_samples": 25000,
6
+ "eval_samples_per_second": 1216.609,
7
+ "eval_steps_per_second": 4.769
8
+ }
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": 1000000000000000019884624838656,
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
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "eval_accuracy": 0.97808,
3
+ "eval_loss": 0.0838116779923439,
4
+ "eval_runtime": 20.6358,
5
+ "eval_samples": 25000,
6
+ "eval_samples_per_second": 1211.49,
7
+ "eval_steps_per_second": 4.749
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,157 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.87396,
3
+ "best_model_checkpoint": "checkpoint/distilbert-base/imdb_42/checkpoint-1564",
4
+ "epoch": 10.0,
5
+ "global_step": 7820,
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
+ "eval_accuracy": 0.87392,
13
+ "eval_loss": 0.29652780294418335,
14
+ "eval_runtime": 8.5888,
15
+ "eval_samples_per_second": 2910.752,
16
+ "eval_steps_per_second": 11.41,
17
+ "step": 782
18
+ },
19
+ {
20
+ "epoch": 1.28,
21
+ "learning_rate": 4.360613810741688e-05,
22
+ "loss": 0.3243,
23
+ "step": 1000
24
+ },
25
+ {
26
+ "epoch": 2.0,
27
+ "eval_accuracy": 0.87396,
28
+ "eval_loss": 0.3133092522621155,
29
+ "eval_runtime": 8.2501,
30
+ "eval_samples_per_second": 3030.277,
31
+ "eval_steps_per_second": 11.879,
32
+ "step": 1564
33
+ },
34
+ {
35
+ "epoch": 2.56,
36
+ "learning_rate": 3.721227621483376e-05,
37
+ "loss": 0.1574,
38
+ "step": 2000
39
+ },
40
+ {
41
+ "epoch": 3.0,
42
+ "eval_accuracy": 0.8712,
43
+ "eval_loss": 0.5505877137184143,
44
+ "eval_runtime": 8.2054,
45
+ "eval_samples_per_second": 3046.769,
46
+ "eval_steps_per_second": 11.943,
47
+ "step": 2346
48
+ },
49
+ {
50
+ "epoch": 3.84,
51
+ "learning_rate": 3.081841432225064e-05,
52
+ "loss": 0.066,
53
+ "step": 3000
54
+ },
55
+ {
56
+ "epoch": 4.0,
57
+ "eval_accuracy": 0.86316,
58
+ "eval_loss": 0.6083658337593079,
59
+ "eval_runtime": 8.1073,
60
+ "eval_samples_per_second": 3083.646,
61
+ "eval_steps_per_second": 12.088,
62
+ "step": 3128
63
+ },
64
+ {
65
+ "epoch": 5.0,
66
+ "eval_accuracy": 0.87084,
67
+ "eval_loss": 0.6640308499336243,
68
+ "eval_runtime": 8.1424,
69
+ "eval_samples_per_second": 3070.331,
70
+ "eval_steps_per_second": 12.036,
71
+ "step": 3910
72
+ },
73
+ {
74
+ "epoch": 5.12,
75
+ "learning_rate": 2.442455242966752e-05,
76
+ "loss": 0.0308,
77
+ "step": 4000
78
+ },
79
+ {
80
+ "epoch": 6.0,
81
+ "eval_accuracy": 0.87112,
82
+ "eval_loss": 0.8268970251083374,
83
+ "eval_runtime": 8.1225,
84
+ "eval_samples_per_second": 3077.853,
85
+ "eval_steps_per_second": 12.065,
86
+ "step": 4692
87
+ },
88
+ {
89
+ "epoch": 6.39,
90
+ "learning_rate": 1.80306905370844e-05,
91
+ "loss": 0.0167,
92
+ "step": 5000
93
+ },
94
+ {
95
+ "epoch": 7.0,
96
+ "eval_accuracy": 0.87168,
97
+ "eval_loss": 0.8184564709663391,
98
+ "eval_runtime": 8.134,
99
+ "eval_samples_per_second": 3073.521,
100
+ "eval_steps_per_second": 12.048,
101
+ "step": 5474
102
+ },
103
+ {
104
+ "epoch": 7.67,
105
+ "learning_rate": 1.163682864450128e-05,
106
+ "loss": 0.0099,
107
+ "step": 6000
108
+ },
109
+ {
110
+ "epoch": 8.0,
111
+ "eval_accuracy": 0.87244,
112
+ "eval_loss": 0.9372827410697937,
113
+ "eval_runtime": 8.0981,
114
+ "eval_samples_per_second": 3087.162,
115
+ "eval_steps_per_second": 12.102,
116
+ "step": 6256
117
+ },
118
+ {
119
+ "epoch": 8.95,
120
+ "learning_rate": 5.242966751918159e-06,
121
+ "loss": 0.0056,
122
+ "step": 7000
123
+ },
124
+ {
125
+ "epoch": 9.0,
126
+ "eval_accuracy": 0.87288,
127
+ "eval_loss": 0.994419276714325,
128
+ "eval_runtime": 8.137,
129
+ "eval_samples_per_second": 3072.372,
130
+ "eval_steps_per_second": 12.044,
131
+ "step": 7038
132
+ },
133
+ {
134
+ "epoch": 10.0,
135
+ "eval_accuracy": 0.87356,
136
+ "eval_loss": 0.9994614720344543,
137
+ "eval_runtime": 8.1462,
138
+ "eval_samples_per_second": 3068.914,
139
+ "eval_steps_per_second": 12.03,
140
+ "step": 7820
141
+ },
142
+ {
143
+ "epoch": 10.0,
144
+ "step": 7820,
145
+ "total_flos": 8279212416000000.0,
146
+ "train_loss": 0.07839516550683609,
147
+ "train_runtime": 418.6292,
148
+ "train_samples_per_second": 597.187,
149
+ "train_steps_per_second": 18.68
150
+ }
151
+ ],
152
+ "max_steps": 7820,
153
+ "num_train_epochs": 10,
154
+ "total_flos": 8279212416000000.0,
155
+ "trial_name": null,
156
+ "trial_params": null
157
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37c29e65f58c26651757d7c6541708c6083f8fd4a3f5bd980beafabc774e8dcc
3
+ size 3631
vocab.txt ADDED
The diff for this file is too large to render. See raw diff