Training in progress, step 100, checkpoint
Browse files- last-checkpoint/config.json +2 -2
- last-checkpoint/model.safetensors +1 -1
- last-checkpoint/optimizer.pt +1 -1
- last-checkpoint/rng_state_0.pth +3 -0
- last-checkpoint/rng_state_1.pth +3 -0
- last-checkpoint/rng_state_2.pth +3 -0
- last-checkpoint/rng_state_3.pth +3 -0
- last-checkpoint/scheduler.pt +1 -1
- last-checkpoint/trainer_state.json +29 -292
- last-checkpoint/training_args.bin +2 -2
last-checkpoint/config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"architectures": [
|
4 |
"EsmForMaskedLM"
|
5 |
],
|
@@ -23,7 +23,7 @@
|
|
23 |
"position_embedding_type": "rotary",
|
24 |
"token_dropout": true,
|
25 |
"torch_dtype": "float32",
|
26 |
-
"transformers_version": "4.
|
27 |
"use_cache": true,
|
28 |
"vocab_list": null,
|
29 |
"vocab_size": 33
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "facebook/esm2_t12_35M_UR50D",
|
3 |
"architectures": [
|
4 |
"EsmForMaskedLM"
|
5 |
],
|
|
|
23 |
"position_embedding_type": "rotary",
|
24 |
"token_dropout": true,
|
25 |
"torch_dtype": "float32",
|
26 |
+
"transformers_version": "4.45.2",
|
27 |
"use_cache": true,
|
28 |
"vocab_list": null,
|
29 |
"vocab_size": 33
|
last-checkpoint/model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 136000488
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:260f33e18a57d6a52a92153fced5f4433b825cbae7e74e13a6bc31c6c3890bb7
|
3 |
size 136000488
|
last-checkpoint/optimizer.pt
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 268176506
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:90a09eabfc08ce01b91af2e622183e0ef79e272762919610e00fff525c6e193b
|
3 |
size 268176506
|
last-checkpoint/rng_state_0.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8dcfee971f88c3cda33b81baee0cbbee275ada59edce084b10f143d24d1066de
|
3 |
+
size 15024
|
last-checkpoint/rng_state_1.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cf7fb54143717feb75da48fb46d5104be02f6922f4165bb1f8dc901d2d29dfe4
|
3 |
+
size 15024
|
last-checkpoint/rng_state_2.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7e20d60dc689485036fab9f46d0c8f9988c61adf50aea162ec6b58003dc5f271
|
3 |
+
size 15024
|
last-checkpoint/rng_state_3.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9cd11604e085d709114e3ae610473be24a56ca9f858b53106ddf99cb2d836153
|
3 |
+
size 15024
|
last-checkpoint/scheduler.pt
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 1064
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:507aa2c5f1b24139884502ecf564787c3e6d78876a1f9bae04804ba26898f238
|
3 |
size 1064
|
last-checkpoint/trainer_state.json
CHANGED
@@ -1,307 +1,44 @@
|
|
1 |
{
|
2 |
-
"best_metric":
|
3 |
-
"best_model_checkpoint":
|
4 |
-
"epoch":
|
5 |
-
"eval_steps":
|
6 |
-
"global_step":
|
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.
|
13 |
-
"grad_norm": 0.
|
14 |
-
"learning_rate":
|
15 |
-
"loss":
|
16 |
-
"step":
|
17 |
},
|
18 |
{
|
19 |
-
"epoch": 0.
|
20 |
-
"
|
21 |
-
"
|
22 |
-
"
|
23 |
-
"
|
24 |
-
|
25 |
-
{
|
26 |
-
"epoch": 0.7226303745634108,
|
27 |
-
"grad_norm": 0.645728349685669,
|
28 |
-
"learning_rate": 3.948413851064344e-06,
|
29 |
-
"loss": 0.7559,
|
30 |
-
"step": 3000
|
31 |
-
},
|
32 |
-
{
|
33 |
-
"epoch": 0.9635071660845478,
|
34 |
-
"grad_norm": 0.6008012294769287,
|
35 |
-
"learning_rate": 3.845884278198482e-06,
|
36 |
-
"loss": 0.7608,
|
37 |
-
"step": 4000
|
38 |
-
},
|
39 |
-
{
|
40 |
-
"epoch": 1.2043839576056847,
|
41 |
-
"grad_norm": 0.7096830606460571,
|
42 |
-
"learning_rate": 3.743354705332619e-06,
|
43 |
-
"loss": 0.7574,
|
44 |
-
"step": 5000
|
45 |
-
},
|
46 |
-
{
|
47 |
-
"epoch": 1.4452607491268217,
|
48 |
-
"grad_norm": 0.5911938548088074,
|
49 |
-
"learning_rate": 3.640825132466757e-06,
|
50 |
-
"loss": 0.7549,
|
51 |
-
"step": 6000
|
52 |
-
},
|
53 |
-
{
|
54 |
-
"epoch": 1.6861375406479586,
|
55 |
-
"grad_norm": 0.7647172212600708,
|
56 |
-
"learning_rate": 3.5382955596008955e-06,
|
57 |
-
"loss": 0.7557,
|
58 |
-
"step": 7000
|
59 |
-
},
|
60 |
-
{
|
61 |
-
"epoch": 1.9270143321690956,
|
62 |
-
"grad_norm": 0.5935486555099487,
|
63 |
-
"learning_rate": 3.435765986735034e-06,
|
64 |
-
"loss": 0.7594,
|
65 |
-
"step": 8000
|
66 |
-
},
|
67 |
-
{
|
68 |
-
"epoch": 2.1678911236902323,
|
69 |
-
"grad_norm": 0.6902479529380798,
|
70 |
-
"learning_rate": 3.333236413869172e-06,
|
71 |
-
"loss": 0.7593,
|
72 |
-
"step": 9000
|
73 |
-
},
|
74 |
-
{
|
75 |
-
"epoch": 2.4087679152113695,
|
76 |
-
"grad_norm": 0.650104284286499,
|
77 |
-
"learning_rate": 3.2307068410033098e-06,
|
78 |
-
"loss": 0.7583,
|
79 |
-
"step": 10000
|
80 |
-
},
|
81 |
-
{
|
82 |
-
"epoch": 2.649644706732506,
|
83 |
-
"grad_norm": 0.5951535701751709,
|
84 |
-
"learning_rate": 3.1281772681374477e-06,
|
85 |
-
"loss": 0.7506,
|
86 |
-
"step": 11000
|
87 |
-
},
|
88 |
-
{
|
89 |
-
"epoch": 2.8905214982536434,
|
90 |
-
"grad_norm": 0.643113911151886,
|
91 |
-
"learning_rate": 3.0256476952715857e-06,
|
92 |
-
"loss": 0.7577,
|
93 |
-
"step": 12000
|
94 |
-
},
|
95 |
-
{
|
96 |
-
"epoch": 3.13139828977478,
|
97 |
-
"grad_norm": 0.5796828866004944,
|
98 |
-
"learning_rate": 2.923118122405724e-06,
|
99 |
-
"loss": 0.7594,
|
100 |
-
"step": 13000
|
101 |
-
},
|
102 |
-
{
|
103 |
-
"epoch": 3.3722750812959172,
|
104 |
-
"grad_norm": 0.6443358063697815,
|
105 |
-
"learning_rate": 2.820588549539862e-06,
|
106 |
-
"loss": 0.7539,
|
107 |
-
"step": 14000
|
108 |
-
},
|
109 |
-
{
|
110 |
-
"epoch": 3.613151872817054,
|
111 |
-
"grad_norm": 0.5989753603935242,
|
112 |
-
"learning_rate": 2.718058976674e-06,
|
113 |
-
"loss": 0.7578,
|
114 |
-
"step": 15000
|
115 |
-
},
|
116 |
-
{
|
117 |
-
"epoch": 3.854028664338191,
|
118 |
-
"grad_norm": 0.5480288863182068,
|
119 |
-
"learning_rate": 2.615529403808138e-06,
|
120 |
-
"loss": 0.7531,
|
121 |
-
"step": 16000
|
122 |
-
},
|
123 |
-
{
|
124 |
-
"epoch": 4.094905455859328,
|
125 |
-
"grad_norm": 0.6654678583145142,
|
126 |
-
"learning_rate": 2.512999830942276e-06,
|
127 |
-
"loss": 0.7579,
|
128 |
-
"step": 17000
|
129 |
-
},
|
130 |
-
{
|
131 |
-
"epoch": 4.335782247380465,
|
132 |
-
"grad_norm": 0.6215384006500244,
|
133 |
-
"learning_rate": 2.4104702580764137e-06,
|
134 |
-
"loss": 0.7545,
|
135 |
-
"step": 18000
|
136 |
-
},
|
137 |
-
{
|
138 |
-
"epoch": 4.576659038901602,
|
139 |
-
"grad_norm": 0.6459180116653442,
|
140 |
-
"learning_rate": 2.3079406852105525e-06,
|
141 |
-
"loss": 0.7566,
|
142 |
-
"step": 19000
|
143 |
-
},
|
144 |
-
{
|
145 |
-
"epoch": 4.817535830422739,
|
146 |
-
"grad_norm": 0.6795477867126465,
|
147 |
-
"learning_rate": 2.2054111123446905e-06,
|
148 |
-
"loss": 0.7555,
|
149 |
-
"step": 20000
|
150 |
-
},
|
151 |
-
{
|
152 |
-
"epoch": 5.058412621943876,
|
153 |
-
"grad_norm": 0.6410796046257019,
|
154 |
-
"learning_rate": 2.1028815394788284e-06,
|
155 |
-
"loss": 0.7503,
|
156 |
-
"step": 21000
|
157 |
-
},
|
158 |
-
{
|
159 |
-
"epoch": 5.299289413465012,
|
160 |
-
"grad_norm": 0.5807074308395386,
|
161 |
-
"learning_rate": 2.0003519666129664e-06,
|
162 |
-
"loss": 0.7529,
|
163 |
-
"step": 22000
|
164 |
-
},
|
165 |
-
{
|
166 |
-
"epoch": 5.54016620498615,
|
167 |
-
"grad_norm": 0.5771316885948181,
|
168 |
-
"learning_rate": 1.8978223937471045e-06,
|
169 |
-
"loss": 0.751,
|
170 |
-
"step": 23000
|
171 |
-
},
|
172 |
-
{
|
173 |
-
"epoch": 5.781042996507287,
|
174 |
-
"grad_norm": 0.6301162838935852,
|
175 |
-
"learning_rate": 1.7952928208812425e-06,
|
176 |
-
"loss": 0.7534,
|
177 |
-
"step": 24000
|
178 |
-
},
|
179 |
-
{
|
180 |
-
"epoch": 6.021919788028423,
|
181 |
-
"grad_norm": 0.5753132104873657,
|
182 |
-
"learning_rate": 1.6927632480153806e-06,
|
183 |
-
"loss": 0.7511,
|
184 |
-
"step": 25000
|
185 |
-
},
|
186 |
-
{
|
187 |
-
"epoch": 6.26279657954956,
|
188 |
-
"grad_norm": 0.6592470407485962,
|
189 |
-
"learning_rate": 1.5902336751495186e-06,
|
190 |
-
"loss": 0.7554,
|
191 |
-
"step": 26000
|
192 |
-
},
|
193 |
-
{
|
194 |
-
"epoch": 6.503673371070697,
|
195 |
-
"grad_norm": 0.6203874349594116,
|
196 |
-
"learning_rate": 1.4877041022836565e-06,
|
197 |
-
"loss": 0.7568,
|
198 |
-
"step": 27000
|
199 |
-
},
|
200 |
-
{
|
201 |
-
"epoch": 6.7445501625918345,
|
202 |
-
"grad_norm": 0.6534298062324524,
|
203 |
-
"learning_rate": 1.385174529417795e-06,
|
204 |
-
"loss": 0.7553,
|
205 |
-
"step": 28000
|
206 |
-
},
|
207 |
-
{
|
208 |
-
"epoch": 6.985426954112971,
|
209 |
-
"grad_norm": 0.6775331497192383,
|
210 |
-
"learning_rate": 1.2826449565519328e-06,
|
211 |
-
"loss": 0.7525,
|
212 |
-
"step": 29000
|
213 |
-
},
|
214 |
-
{
|
215 |
-
"epoch": 7.226303745634108,
|
216 |
-
"grad_norm": 0.640843391418457,
|
217 |
-
"learning_rate": 1.180115383686071e-06,
|
218 |
-
"loss": 0.7559,
|
219 |
-
"step": 30000
|
220 |
-
},
|
221 |
-
{
|
222 |
-
"epoch": 7.4671805371552455,
|
223 |
-
"grad_norm": 0.6803774833679199,
|
224 |
-
"learning_rate": 1.077585810820209e-06,
|
225 |
-
"loss": 0.7508,
|
226 |
-
"step": 31000
|
227 |
-
},
|
228 |
-
{
|
229 |
-
"epoch": 7.708057328676382,
|
230 |
-
"grad_norm": 0.6412643790245056,
|
231 |
-
"learning_rate": 9.75056237954347e-07,
|
232 |
-
"loss": 0.7534,
|
233 |
-
"step": 32000
|
234 |
-
},
|
235 |
-
{
|
236 |
-
"epoch": 7.948934120197519,
|
237 |
-
"grad_norm": 0.6142516136169434,
|
238 |
-
"learning_rate": 8.725266650884852e-07,
|
239 |
-
"loss": 0.7531,
|
240 |
-
"step": 33000
|
241 |
-
},
|
242 |
-
{
|
243 |
-
"epoch": 8.189810911718656,
|
244 |
-
"grad_norm": 0.670280933380127,
|
245 |
-
"learning_rate": 7.699970922226231e-07,
|
246 |
-
"loss": 0.7506,
|
247 |
-
"step": 34000
|
248 |
-
},
|
249 |
-
{
|
250 |
-
"epoch": 8.430687703239792,
|
251 |
-
"grad_norm": 0.6336104869842529,
|
252 |
-
"learning_rate": 6.674675193567612e-07,
|
253 |
-
"loss": 0.7538,
|
254 |
-
"step": 35000
|
255 |
-
},
|
256 |
-
{
|
257 |
-
"epoch": 8.67156449476093,
|
258 |
-
"grad_norm": 0.6225976347923279,
|
259 |
-
"learning_rate": 5.649379464908993e-07,
|
260 |
-
"loss": 0.7521,
|
261 |
-
"step": 36000
|
262 |
-
},
|
263 |
-
{
|
264 |
-
"epoch": 8.912441286282068,
|
265 |
-
"grad_norm": 0.7998696565628052,
|
266 |
-
"learning_rate": 4.624083736250373e-07,
|
267 |
-
"loss": 0.7559,
|
268 |
-
"step": 37000
|
269 |
-
},
|
270 |
-
{
|
271 |
-
"epoch": 9.153318077803204,
|
272 |
-
"grad_norm": 0.6381024122238159,
|
273 |
-
"learning_rate": 3.598788007591754e-07,
|
274 |
-
"loss": 0.7547,
|
275 |
-
"step": 38000
|
276 |
-
},
|
277 |
-
{
|
278 |
-
"epoch": 9.394194869324341,
|
279 |
-
"grad_norm": 0.558660089969635,
|
280 |
-
"learning_rate": 2.5734922789331346e-07,
|
281 |
-
"loss": 0.7539,
|
282 |
-
"step": 39000
|
283 |
-
},
|
284 |
-
{
|
285 |
-
"epoch": 9.635071660845478,
|
286 |
-
"grad_norm": 0.7711436748504639,
|
287 |
-
"learning_rate": 1.5481965502745152e-07,
|
288 |
-
"loss": 0.7484,
|
289 |
-
"step": 40000
|
290 |
-
},
|
291 |
-
{
|
292 |
-
"epoch": 9.875948452366615,
|
293 |
-
"grad_norm": 0.6147646903991699,
|
294 |
-
"learning_rate": 5.229008216158959e-08,
|
295 |
-
"loss": 0.7509,
|
296 |
-
"step": 41000
|
297 |
}
|
298 |
],
|
299 |
-
"logging_steps":
|
300 |
-
"max_steps":
|
301 |
"num_input_tokens_seen": 0,
|
302 |
"num_train_epochs": 10,
|
303 |
-
"save_steps":
|
304 |
"stateful_callbacks": {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
305 |
"TrainerControl": {
|
306 |
"args": {
|
307 |
"should_epoch_stop": false,
|
@@ -313,7 +50,7 @@
|
|
313 |
"attributes": {}
|
314 |
}
|
315 |
},
|
316 |
-
"total_flos":
|
317 |
"train_batch_size": 64,
|
318 |
"trial_name": null,
|
319 |
"trial_params": null
|
|
|
1 |
{
|
2 |
+
"best_metric": 1.250386118888855,
|
3 |
+
"best_model_checkpoint": "mgh6/TCS_MLM_All/checkpoint-100",
|
4 |
+
"epoch": 0.9523809523809523,
|
5 |
+
"eval_steps": 100,
|
6 |
+
"global_step": 100,
|
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.9523809523809523,
|
13 |
+
"grad_norm": 0.18039299547672272,
|
14 |
+
"learning_rate": 0.0009047619047619047,
|
15 |
+
"loss": 1.5254,
|
16 |
+
"step": 100
|
17 |
},
|
18 |
{
|
19 |
+
"epoch": 0.9523809523809523,
|
20 |
+
"eval_loss": 1.250386118888855,
|
21 |
+
"eval_runtime": 0.218,
|
22 |
+
"eval_samples_per_second": 87.168,
|
23 |
+
"eval_steps_per_second": 4.588,
|
24 |
+
"step": 100
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
}
|
26 |
],
|
27 |
+
"logging_steps": 100,
|
28 |
+
"max_steps": 1050,
|
29 |
"num_input_tokens_seen": 0,
|
30 |
"num_train_epochs": 10,
|
31 |
+
"save_steps": 100,
|
32 |
"stateful_callbacks": {
|
33 |
+
"EarlyStoppingCallback": {
|
34 |
+
"args": {
|
35 |
+
"early_stopping_patience": 5,
|
36 |
+
"early_stopping_threshold": 0.0
|
37 |
+
},
|
38 |
+
"attributes": {
|
39 |
+
"early_stopping_patience_counter": 0
|
40 |
+
}
|
41 |
+
},
|
42 |
"TrainerControl": {
|
43 |
"args": {
|
44 |
"should_epoch_stop": false,
|
|
|
50 |
"attributes": {}
|
51 |
}
|
52 |
},
|
53 |
+
"total_flos": 2931727227944960.0,
|
54 |
"train_batch_size": 64,
|
55 |
"trial_name": null,
|
56 |
"trial_params": null
|
last-checkpoint/training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:873e2b625b704f36c39ea8938bc18e37074ce987126ad27c9b364f4977d71919
|
3 |
+
size 5240
|