simonloewe commited on
Commit
92b9f1a
·
1 Parent(s): 392fb1e

First stigma model

Browse files
Files changed (7) hide show
  1. config.json +28 -0
  2. optimizer.pt +3 -0
  3. pytorch_model.bin +3 -0
  4. rng_state.pth +3 -0
  5. scheduler.pt +3 -0
  6. trainer_state.json +184 -0
  7. training_args.bin +3 -0
config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 514,
17
+ "model_type": "roberta",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 6,
20
+ "pad_token_id": 1,
21
+ "position_embedding_type": "absolute",
22
+ "problem_type": "single_label_classification",
23
+ "torch_dtype": "float32",
24
+ "transformers_version": "4.28.1",
25
+ "type_vocab_size": 1,
26
+ "use_cache": true,
27
+ "vocab_size": 50265
28
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:75c82fcc45490ded6f9b7db15e39541f26579e1ccb85d9ebd3ceef9446d20148
3
+ size 657020933
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:073f6035a2e1fd67e1389e9d081614431d1c6f739c7ffc13a9f5c4491223a36f
3
+ size 328519925
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:117a95bd0514634aa2e66194a1b64fb0f2c38ea2af7c2410bfd584c58dd20964
3
+ size 14575
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:46b0e38e86e2b0ae8fcc43efeaae364ad84828772c63c0375ed9e8635129b7f2
3
+ size 627
trainer_state.json ADDED
@@ -0,0 +1,184 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.6666666666666667,
3
+ "best_model_checkpoint": "NLP/trained_models/256_5e-06_90/checkpoint-140",
4
+ "epoch": 70.0,
5
+ "global_step": 140,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 5.0,
12
+ "eval_accuracy": 0.5647058823529412,
13
+ "eval_f1": 0.0,
14
+ "eval_loss": 0.6881718635559082,
15
+ "eval_precision": 0.0,
16
+ "eval_recall": 0.0,
17
+ "eval_runtime": 2.0353,
18
+ "eval_samples_per_second": 41.763,
19
+ "eval_steps_per_second": 0.491,
20
+ "step": 10
21
+ },
22
+ {
23
+ "epoch": 10.0,
24
+ "eval_accuracy": 0.5647058823529412,
25
+ "eval_f1": 0.0,
26
+ "eval_loss": 0.6862292885780334,
27
+ "eval_precision": 0.0,
28
+ "eval_recall": 0.0,
29
+ "eval_runtime": 3.3692,
30
+ "eval_samples_per_second": 25.229,
31
+ "eval_steps_per_second": 0.297,
32
+ "step": 20
33
+ },
34
+ {
35
+ "epoch": 15.0,
36
+ "eval_accuracy": 0.5764705882352941,
37
+ "eval_f1": 0.1,
38
+ "eval_loss": 0.6834429502487183,
39
+ "eval_precision": 0.6666666666666666,
40
+ "eval_recall": 0.05405405405405406,
41
+ "eval_runtime": 1.8134,
42
+ "eval_samples_per_second": 46.873,
43
+ "eval_steps_per_second": 0.551,
44
+ "step": 30
45
+ },
46
+ {
47
+ "epoch": 20.0,
48
+ "eval_accuracy": 0.6235294117647059,
49
+ "eval_f1": 0.3846153846153846,
50
+ "eval_loss": 0.6755977272987366,
51
+ "eval_precision": 0.6666666666666666,
52
+ "eval_recall": 0.2702702702702703,
53
+ "eval_runtime": 2.1098,
54
+ "eval_samples_per_second": 40.288,
55
+ "eval_steps_per_second": 0.474,
56
+ "step": 40
57
+ },
58
+ {
59
+ "epoch": 25.0,
60
+ "eval_accuracy": 0.611764705882353,
61
+ "eval_f1": 0.47619047619047616,
62
+ "eval_loss": 0.6568019390106201,
63
+ "eval_precision": 0.5769230769230769,
64
+ "eval_recall": 0.40540540540540543,
65
+ "eval_runtime": 2.4655,
66
+ "eval_samples_per_second": 34.475,
67
+ "eval_steps_per_second": 0.406,
68
+ "step": 50
69
+ },
70
+ {
71
+ "epoch": 30.0,
72
+ "eval_accuracy": 0.6588235294117647,
73
+ "eval_f1": 0.5797101449275363,
74
+ "eval_loss": 0.6438259482383728,
75
+ "eval_precision": 0.625,
76
+ "eval_recall": 0.5405405405405406,
77
+ "eval_runtime": 1.8313,
78
+ "eval_samples_per_second": 46.416,
79
+ "eval_steps_per_second": 0.546,
80
+ "step": 60
81
+ },
82
+ {
83
+ "epoch": 35.0,
84
+ "eval_accuracy": 0.6588235294117647,
85
+ "eval_f1": 0.6133333333333333,
86
+ "eval_loss": 0.6628918647766113,
87
+ "eval_precision": 0.6052631578947368,
88
+ "eval_recall": 0.6216216216216216,
89
+ "eval_runtime": 1.8199,
90
+ "eval_samples_per_second": 46.706,
91
+ "eval_steps_per_second": 0.549,
92
+ "step": 70
93
+ },
94
+ {
95
+ "epoch": 40.0,
96
+ "eval_accuracy": 0.6588235294117647,
97
+ "eval_f1": 0.6133333333333333,
98
+ "eval_loss": 0.6717644333839417,
99
+ "eval_precision": 0.6052631578947368,
100
+ "eval_recall": 0.6216216216216216,
101
+ "eval_runtime": 6.0426,
102
+ "eval_samples_per_second": 14.067,
103
+ "eval_steps_per_second": 0.165,
104
+ "step": 80
105
+ },
106
+ {
107
+ "epoch": 45.0,
108
+ "eval_accuracy": 0.6823529411764706,
109
+ "eval_f1": 0.6666666666666666,
110
+ "eval_loss": 0.7056936025619507,
111
+ "eval_precision": 0.6136363636363636,
112
+ "eval_recall": 0.7297297297297297,
113
+ "eval_runtime": 4.3129,
114
+ "eval_samples_per_second": 19.708,
115
+ "eval_steps_per_second": 0.232,
116
+ "step": 90
117
+ },
118
+ {
119
+ "epoch": 50.0,
120
+ "eval_accuracy": 0.6705882352941176,
121
+ "eval_f1": 0.6410256410256411,
122
+ "eval_loss": 0.7082163095474243,
123
+ "eval_precision": 0.6097560975609756,
124
+ "eval_recall": 0.6756756756756757,
125
+ "eval_runtime": 2.7778,
126
+ "eval_samples_per_second": 30.6,
127
+ "eval_steps_per_second": 0.36,
128
+ "step": 100
129
+ },
130
+ {
131
+ "epoch": 55.0,
132
+ "eval_accuracy": 0.6705882352941176,
133
+ "eval_f1": 0.6499999999999999,
134
+ "eval_loss": 0.7371630668640137,
135
+ "eval_precision": 0.6046511627906976,
136
+ "eval_recall": 0.7027027027027027,
137
+ "eval_runtime": 2.1835,
138
+ "eval_samples_per_second": 38.928,
139
+ "eval_steps_per_second": 0.458,
140
+ "step": 110
141
+ },
142
+ {
143
+ "epoch": 60.0,
144
+ "eval_accuracy": 0.6823529411764706,
145
+ "eval_f1": 0.6582278481012659,
146
+ "eval_loss": 0.7530251145362854,
147
+ "eval_precision": 0.6190476190476191,
148
+ "eval_recall": 0.7027027027027027,
149
+ "eval_runtime": 2.1705,
150
+ "eval_samples_per_second": 39.161,
151
+ "eval_steps_per_second": 0.461,
152
+ "step": 120
153
+ },
154
+ {
155
+ "epoch": 65.0,
156
+ "eval_accuracy": 0.6823529411764706,
157
+ "eval_f1": 0.6582278481012659,
158
+ "eval_loss": 0.7745981812477112,
159
+ "eval_precision": 0.6190476190476191,
160
+ "eval_recall": 0.7027027027027027,
161
+ "eval_runtime": 2.8635,
162
+ "eval_samples_per_second": 29.684,
163
+ "eval_steps_per_second": 0.349,
164
+ "step": 130
165
+ },
166
+ {
167
+ "epoch": 70.0,
168
+ "eval_accuracy": 0.6941176470588235,
169
+ "eval_f1": 0.6666666666666667,
170
+ "eval_loss": 0.7838086485862732,
171
+ "eval_precision": 0.6341463414634146,
172
+ "eval_recall": 0.7027027027027027,
173
+ "eval_runtime": 1.9223,
174
+ "eval_samples_per_second": 44.219,
175
+ "eval_steps_per_second": 0.52,
176
+ "step": 140
177
+ }
178
+ ],
179
+ "max_steps": 180,
180
+ "num_train_epochs": 90,
181
+ "total_flos": 116488518693120.0,
182
+ "trial_name": null,
183
+ "trial_params": null
184
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3123bb444fcb1096aa3a773774c3c9eee8cb9d6331b894f1ec8953f09692681
3
+ size 3579