jialicheng commited on
Commit
702db90
1 Parent(s): c2ca9c8

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: google/electra-base-discriminator
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 [google/electra-base-discriminator](https://huggingface.co/google/electra-base-discriminator) on an unknown dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - eval_loss: 0.0027
19
+ - eval_accuracy: 0.9995
20
+ - eval_runtime: 40.0833
21
+ - eval_samples_per_second: 623.7
22
+ - eval_steps_per_second: 2.445
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.99948,
4
+ "eval_loss": 0.002695722272619605,
5
+ "eval_runtime": 40.0833,
6
+ "eval_samples": 25000,
7
+ "eval_samples_per_second": 623.7,
8
+ "eval_steps_per_second": 2.445,
9
+ "train_loss": 0.0733462499535602,
10
+ "train_runtime": 793.1279,
11
+ "train_samples": 25000,
12
+ "train_samples_per_second": 315.208,
13
+ "train_steps_per_second": 9.86
14
+ }
config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/electra-base-discriminator",
3
+ "architectures": [
4
+ "ElectraForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "embedding_size": 768,
9
+ "finetuning_task": "text-classification",
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "id2label": {
14
+ "0": "0",
15
+ "1": "1"
16
+ },
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 3072,
19
+ "label2id": {
20
+ "0": 0,
21
+ "1": 1
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
+ "max_position_embeddings": 512,
25
+ "model_type": "electra",
26
+ "num_attention_heads": 12,
27
+ "num_hidden_layers": 12,
28
+ "pad_token_id": 0,
29
+ "position_embedding_type": "absolute",
30
+ "problem_type": "single_label_classification",
31
+ "summary_activation": "gelu",
32
+ "summary_last_dropout": 0.1,
33
+ "summary_type": "first",
34
+ "summary_use_proj": true,
35
+ "torch_dtype": "float32",
36
+ "transformers_version": "4.28.0",
37
+ "type_vocab_size": 2,
38
+ "use_cache": true,
39
+ "vocab_size": 30522
40
+ }
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "eval_accuracy": 0.90888,
3
+ "eval_loss": 0.6276622414588928,
4
+ "eval_runtime": 44.1174,
5
+ "eval_samples": 25000,
6
+ "eval_samples_per_second": 566.67,
7
+ "eval_steps_per_second": 2.221
8
+ }
pred_logit_eval.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:462690d0106a3ac59092b5b3a95bc93f2bdb2db684db4e9ad25416ad812572aa
3
+ size 200128
pred_logit_test.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:462690d0106a3ac59092b5b3a95bc93f2bdb2db684db4e9ad25416ad812572aa
3
+ size 200128
pred_logit_train.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d00b0f191ba17c3cb47d8fa13ceba382c1aeb4183311e580c287a6073885aee
3
+ size 200128
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:044ec9f99f751142ee318e6cff4ab49c5ab0c7c2455b377644a4825cd8d63419
3
+ size 438006701
runs/Apr17_12-28-18_clu/events.out.tfevents.1713356970.clu ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d98b97b7517fdd4dff02920dbb204f17a40da398a3d47115919e30bd5e7ca45
3
+ size 298
runs/Apr21_18-55-07_clu/events.out.tfevents.1713725769.clu ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbabac543571e86c1f5f645e58693a8e41fc94e72952f3a2f0bcb443a306422a
3
+ size 814
runs/Mar18_11-35-05_clu/1710761732.41874/events.out.tfevents.1710761732.clu ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d69614210e6265f8c4dd867dc18239a0586a04c0fe575404da64826d34c3fdbf
3
+ size 5631
runs/Mar18_11-35-05_clu/events.out.tfevents.1710761732.clu ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e40ba1bf3800bbdadac89f8726096d06c5c36a3302e0b1882f23c4c58b22e91
3
+ size 9007
runs/Mar18_11-35-05_clu/events.out.tfevents.1710762541.clu ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c46626d424196aa49c3efae4b9636c221bd0a14dfedb1129042a65edb26ccff
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.90888,
3
+ "eval_loss": 0.6276622414588928,
4
+ "eval_runtime": 40.0814,
5
+ "eval_samples": 25000,
6
+ "eval_samples_per_second": 623.73,
7
+ "eval_steps_per_second": 2.445
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": 512,
7
+ "pad_token": "[PAD]",
8
+ "sep_token": "[SEP]",
9
+ "strip_accents": null,
10
+ "tokenize_chinese_chars": true,
11
+ "tokenizer_class": "ElectraTokenizer",
12
+ "unk_token": "[UNK]"
13
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "eval_accuracy": 0.99948,
3
+ "eval_loss": 0.002695722272619605,
4
+ "eval_runtime": 40.0833,
5
+ "eval_samples": 25000,
6
+ "eval_samples_per_second": 623.7,
7
+ "eval_steps_per_second": 2.445
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,157 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.90888,
3
+ "best_model_checkpoint": "checkpoint/electra-base/imdb_42/checkpoint-7038",
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.90536,
13
+ "eval_loss": 0.23193472623825073,
14
+ "eval_runtime": 14.9471,
15
+ "eval_samples_per_second": 1672.56,
16
+ "eval_steps_per_second": 6.556,
17
+ "step": 782
18
+ },
19
+ {
20
+ "epoch": 1.28,
21
+ "learning_rate": 4.360613810741688e-05,
22
+ "loss": 0.2699,
23
+ "step": 1000
24
+ },
25
+ {
26
+ "epoch": 2.0,
27
+ "eval_accuracy": 0.90432,
28
+ "eval_loss": 0.2524752616882324,
29
+ "eval_runtime": 14.5935,
30
+ "eval_samples_per_second": 1713.094,
31
+ "eval_steps_per_second": 6.715,
32
+ "step": 1564
33
+ },
34
+ {
35
+ "epoch": 2.56,
36
+ "learning_rate": 3.721227621483376e-05,
37
+ "loss": 0.1432,
38
+ "step": 2000
39
+ },
40
+ {
41
+ "epoch": 3.0,
42
+ "eval_accuracy": 0.90284,
43
+ "eval_loss": 0.4573332369327545,
44
+ "eval_runtime": 14.9943,
45
+ "eval_samples_per_second": 1667.297,
46
+ "eval_steps_per_second": 6.536,
47
+ "step": 2346
48
+ },
49
+ {
50
+ "epoch": 3.84,
51
+ "learning_rate": 3.081841432225064e-05,
52
+ "loss": 0.072,
53
+ "step": 3000
54
+ },
55
+ {
56
+ "epoch": 4.0,
57
+ "eval_accuracy": 0.8994,
58
+ "eval_loss": 0.411539226770401,
59
+ "eval_runtime": 14.9739,
60
+ "eval_samples_per_second": 1669.577,
61
+ "eval_steps_per_second": 6.545,
62
+ "step": 3128
63
+ },
64
+ {
65
+ "epoch": 5.0,
66
+ "eval_accuracy": 0.9038,
67
+ "eval_loss": 0.45995330810546875,
68
+ "eval_runtime": 14.943,
69
+ "eval_samples_per_second": 1673.022,
70
+ "eval_steps_per_second": 6.558,
71
+ "step": 3910
72
+ },
73
+ {
74
+ "epoch": 5.12,
75
+ "learning_rate": 2.442455242966752e-05,
76
+ "loss": 0.0384,
77
+ "step": 4000
78
+ },
79
+ {
80
+ "epoch": 6.0,
81
+ "eval_accuracy": 0.9056,
82
+ "eval_loss": 0.47831761837005615,
83
+ "eval_runtime": 15.0585,
84
+ "eval_samples_per_second": 1660.194,
85
+ "eval_steps_per_second": 6.508,
86
+ "step": 4692
87
+ },
88
+ {
89
+ "epoch": 6.39,
90
+ "learning_rate": 1.80306905370844e-05,
91
+ "loss": 0.0227,
92
+ "step": 5000
93
+ },
94
+ {
95
+ "epoch": 7.0,
96
+ "eval_accuracy": 0.90184,
97
+ "eval_loss": 0.5771753787994385,
98
+ "eval_runtime": 15.0276,
99
+ "eval_samples_per_second": 1663.606,
100
+ "eval_steps_per_second": 6.521,
101
+ "step": 5474
102
+ },
103
+ {
104
+ "epoch": 7.67,
105
+ "learning_rate": 1.163682864450128e-05,
106
+ "loss": 0.0144,
107
+ "step": 6000
108
+ },
109
+ {
110
+ "epoch": 8.0,
111
+ "eval_accuracy": 0.90868,
112
+ "eval_loss": 0.5959281921386719,
113
+ "eval_runtime": 14.9693,
114
+ "eval_samples_per_second": 1670.087,
115
+ "eval_steps_per_second": 6.547,
116
+ "step": 6256
117
+ },
118
+ {
119
+ "epoch": 8.95,
120
+ "learning_rate": 5.242966751918159e-06,
121
+ "loss": 0.01,
122
+ "step": 7000
123
+ },
124
+ {
125
+ "epoch": 9.0,
126
+ "eval_accuracy": 0.90888,
127
+ "eval_loss": 0.6276312470436096,
128
+ "eval_runtime": 14.9974,
129
+ "eval_samples_per_second": 1666.955,
130
+ "eval_steps_per_second": 6.534,
131
+ "step": 7038
132
+ },
133
+ {
134
+ "epoch": 10.0,
135
+ "eval_accuracy": 0.90888,
136
+ "eval_loss": 0.6741125583648682,
137
+ "eval_runtime": 14.9871,
138
+ "eval_samples_per_second": 1668.106,
139
+ "eval_steps_per_second": 6.539,
140
+ "step": 7820
141
+ },
142
+ {
143
+ "epoch": 10.0,
144
+ "step": 7820,
145
+ "total_flos": 1.644444096e+16,
146
+ "train_loss": 0.0733462499535602,
147
+ "train_runtime": 793.1279,
148
+ "train_samples_per_second": 315.208,
149
+ "train_steps_per_second": 9.86
150
+ }
151
+ ],
152
+ "max_steps": 7820,
153
+ "num_train_epochs": 10,
154
+ "total_flos": 1.644444096e+16,
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:cff0ef9332433f8dd6ce58dc2c676ee07395b285992da1808e5db997e35d1647
3
+ size 3631
vocab.txt ADDED
The diff for this file is too large to render. See raw diff