AmelieSchreiber commited on
Commit
48331b0
·
1 Parent(s): bde6e05

Upload 7 files

Browse files
config (6).json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/esm2_t33_650M_UR50D",
3
+ "architectures": [
4
+ "EsmForMaskedLM"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout": null,
8
+ "emb_layer_norm_before": false,
9
+ "esmfold_config": null,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.0,
12
+ "hidden_size": 1280,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 5120,
15
+ "is_folding_model": false,
16
+ "layer_norm_eps": 1e-05,
17
+ "mask_token_id": 32,
18
+ "max_position_embeddings": 1026,
19
+ "model_type": "esm",
20
+ "num_attention_heads": 20,
21
+ "num_hidden_layers": 33,
22
+ "pad_token_id": 1,
23
+ "position_embedding_type": "rotary",
24
+ "token_dropout": true,
25
+ "torch_dtype": "float32",
26
+ "transformers_version": "4.35.2",
27
+ "use_cache": true,
28
+ "vocab_list": null,
29
+ "vocab_size": 33
30
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61fd44ef1c034eb4739b499daff54a586d84cb4064e8200c74ea18f34c04fc01
3
+ size 2609498088
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf189f76489a8d7251058c21008aaa264ba0a9f54cd97b71dc6f89024ace4d92
3
+ size 5208792737
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6821b6ab47ebf2687f3d998057cb9a617e892c9cb0db758974ebcc0c1ea5471
3
+ size 14575
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b24bf7d361c8b88fb3078a19658f349617d98e2b5e6b3840ddd113ec2de5de5
3
+ size 563
trainer_state.json ADDED
@@ -0,0 +1,581 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.3603260517120361,
3
+ "best_model_checkpoint": "./interact_output/checkpoint-914",
4
+ "epoch": 2.0,
5
+ "eval_steps": 500,
6
+ "global_step": 914,
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.02,
13
+ "learning_rate": 0.00015968553632342872,
14
+ "loss": 1.8187,
15
+ "step": 10
16
+ },
17
+ {
18
+ "epoch": 0.04,
19
+ "learning_rate": 0.00031937107264685743,
20
+ "loss": 1.7865,
21
+ "step": 20
22
+ },
23
+ {
24
+ "epoch": 0.07,
25
+ "learning_rate": 0.00047905660897028615,
26
+ "loss": 1.9059,
27
+ "step": 30
28
+ },
29
+ {
30
+ "epoch": 0.09,
31
+ "learning_rate": 0.0006387421452937149,
32
+ "loss": 1.9266,
33
+ "step": 40
34
+ },
35
+ {
36
+ "epoch": 0.11,
37
+ "learning_rate": 0.0007984276816171436,
38
+ "loss": 1.9969,
39
+ "step": 50
40
+ },
41
+ {
42
+ "epoch": 0.13,
43
+ "learning_rate": 0.0007984276816171436,
44
+ "loss": 2.0468,
45
+ "step": 60
46
+ },
47
+ {
48
+ "epoch": 0.15,
49
+ "learning_rate": 0.0007984276816171436,
50
+ "loss": 1.9863,
51
+ "step": 70
52
+ },
53
+ {
54
+ "epoch": 0.18,
55
+ "learning_rate": 0.0007984276816171436,
56
+ "loss": 1.8926,
57
+ "step": 80
58
+ },
59
+ {
60
+ "epoch": 0.2,
61
+ "learning_rate": 0.0007984276816171436,
62
+ "loss": 1.8434,
63
+ "step": 90
64
+ },
65
+ {
66
+ "epoch": 0.22,
67
+ "learning_rate": 0.0007984276816171436,
68
+ "loss": 1.8316,
69
+ "step": 100
70
+ },
71
+ {
72
+ "epoch": 0.24,
73
+ "learning_rate": 0.0007984276816171436,
74
+ "loss": 1.7679,
75
+ "step": 110
76
+ },
77
+ {
78
+ "epoch": 0.26,
79
+ "learning_rate": 0.0007984276816171436,
80
+ "loss": 1.7136,
81
+ "step": 120
82
+ },
83
+ {
84
+ "epoch": 0.28,
85
+ "learning_rate": 0.0007984276816171436,
86
+ "loss": 1.8007,
87
+ "step": 130
88
+ },
89
+ {
90
+ "epoch": 0.31,
91
+ "learning_rate": 0.0007984276816171436,
92
+ "loss": 1.7397,
93
+ "step": 140
94
+ },
95
+ {
96
+ "epoch": 0.33,
97
+ "learning_rate": 0.0007984276816171436,
98
+ "loss": 1.7168,
99
+ "step": 150
100
+ },
101
+ {
102
+ "epoch": 0.35,
103
+ "learning_rate": 0.0007984276816171436,
104
+ "loss": 1.7507,
105
+ "step": 160
106
+ },
107
+ {
108
+ "epoch": 0.37,
109
+ "learning_rate": 0.0007984276816171436,
110
+ "loss": 1.5584,
111
+ "step": 170
112
+ },
113
+ {
114
+ "epoch": 0.39,
115
+ "learning_rate": 0.0007984276816171436,
116
+ "loss": 1.7085,
117
+ "step": 180
118
+ },
119
+ {
120
+ "epoch": 0.42,
121
+ "learning_rate": 0.0007984276816171436,
122
+ "loss": 1.6188,
123
+ "step": 190
124
+ },
125
+ {
126
+ "epoch": 0.44,
127
+ "learning_rate": 0.0007984276816171436,
128
+ "loss": 1.5501,
129
+ "step": 200
130
+ },
131
+ {
132
+ "epoch": 0.46,
133
+ "learning_rate": 0.0007984276816171436,
134
+ "loss": 1.6778,
135
+ "step": 210
136
+ },
137
+ {
138
+ "epoch": 0.48,
139
+ "learning_rate": 0.0007984276816171436,
140
+ "loss": 1.6112,
141
+ "step": 220
142
+ },
143
+ {
144
+ "epoch": 0.5,
145
+ "learning_rate": 0.0007984276816171436,
146
+ "loss": 1.5693,
147
+ "step": 230
148
+ },
149
+ {
150
+ "epoch": 0.53,
151
+ "learning_rate": 0.0007984276816171436,
152
+ "loss": 1.5671,
153
+ "step": 240
154
+ },
155
+ {
156
+ "epoch": 0.55,
157
+ "learning_rate": 0.0007984276816171436,
158
+ "loss": 1.5477,
159
+ "step": 250
160
+ },
161
+ {
162
+ "epoch": 0.57,
163
+ "learning_rate": 0.0007984276816171436,
164
+ "loss": 1.5861,
165
+ "step": 260
166
+ },
167
+ {
168
+ "epoch": 0.59,
169
+ "learning_rate": 0.0007984276816171436,
170
+ "loss": 1.6772,
171
+ "step": 270
172
+ },
173
+ {
174
+ "epoch": 0.61,
175
+ "learning_rate": 0.0007984276816171436,
176
+ "loss": 1.5901,
177
+ "step": 280
178
+ },
179
+ {
180
+ "epoch": 0.63,
181
+ "learning_rate": 0.0007984276816171436,
182
+ "loss": 1.5628,
183
+ "step": 290
184
+ },
185
+ {
186
+ "epoch": 0.66,
187
+ "learning_rate": 0.0007984276816171436,
188
+ "loss": 1.5902,
189
+ "step": 300
190
+ },
191
+ {
192
+ "epoch": 0.68,
193
+ "learning_rate": 0.0007984276816171436,
194
+ "loss": 1.4775,
195
+ "step": 310
196
+ },
197
+ {
198
+ "epoch": 0.7,
199
+ "learning_rate": 0.0007984276816171436,
200
+ "loss": 1.5247,
201
+ "step": 320
202
+ },
203
+ {
204
+ "epoch": 0.72,
205
+ "learning_rate": 0.0007984276816171436,
206
+ "loss": 1.5076,
207
+ "step": 330
208
+ },
209
+ {
210
+ "epoch": 0.74,
211
+ "learning_rate": 0.0007984276816171436,
212
+ "loss": 1.483,
213
+ "step": 340
214
+ },
215
+ {
216
+ "epoch": 0.77,
217
+ "learning_rate": 0.0007984276816171436,
218
+ "loss": 1.5376,
219
+ "step": 350
220
+ },
221
+ {
222
+ "epoch": 0.79,
223
+ "learning_rate": 0.0007984276816171436,
224
+ "loss": 1.4295,
225
+ "step": 360
226
+ },
227
+ {
228
+ "epoch": 0.81,
229
+ "learning_rate": 0.0007984276816171436,
230
+ "loss": 1.4517,
231
+ "step": 370
232
+ },
233
+ {
234
+ "epoch": 0.83,
235
+ "learning_rate": 0.0007984276816171436,
236
+ "loss": 1.5043,
237
+ "step": 380
238
+ },
239
+ {
240
+ "epoch": 0.85,
241
+ "learning_rate": 0.0007984276816171436,
242
+ "loss": 1.3862,
243
+ "step": 390
244
+ },
245
+ {
246
+ "epoch": 0.88,
247
+ "learning_rate": 0.0007984276816171436,
248
+ "loss": 1.4554,
249
+ "step": 400
250
+ },
251
+ {
252
+ "epoch": 0.9,
253
+ "learning_rate": 0.0007984276816171436,
254
+ "loss": 1.417,
255
+ "step": 410
256
+ },
257
+ {
258
+ "epoch": 0.92,
259
+ "learning_rate": 0.0007984276816171436,
260
+ "loss": 1.4428,
261
+ "step": 420
262
+ },
263
+ {
264
+ "epoch": 0.94,
265
+ "learning_rate": 0.0007984276816171436,
266
+ "loss": 1.4553,
267
+ "step": 430
268
+ },
269
+ {
270
+ "epoch": 0.96,
271
+ "learning_rate": 0.0007984276816171436,
272
+ "loss": 1.4517,
273
+ "step": 440
274
+ },
275
+ {
276
+ "epoch": 0.98,
277
+ "learning_rate": 0.0007984276816171436,
278
+ "loss": 1.4568,
279
+ "step": 450
280
+ },
281
+ {
282
+ "epoch": 1.0,
283
+ "eval_loss": 1.5652484893798828,
284
+ "eval_runtime": 208.8087,
285
+ "eval_samples_per_second": 12.902,
286
+ "eval_steps_per_second": 3.228,
287
+ "step": 457
288
+ },
289
+ {
290
+ "epoch": 1.01,
291
+ "learning_rate": 0.0007984276816171436,
292
+ "loss": 1.3963,
293
+ "step": 460
294
+ },
295
+ {
296
+ "epoch": 1.03,
297
+ "learning_rate": 0.0007984276816171436,
298
+ "loss": 1.3257,
299
+ "step": 470
300
+ },
301
+ {
302
+ "epoch": 1.05,
303
+ "learning_rate": 0.0007984276816171436,
304
+ "loss": 1.2477,
305
+ "step": 480
306
+ },
307
+ {
308
+ "epoch": 1.07,
309
+ "learning_rate": 0.0007984276816171436,
310
+ "loss": 1.283,
311
+ "step": 490
312
+ },
313
+ {
314
+ "epoch": 1.09,
315
+ "learning_rate": 0.0007984276816171436,
316
+ "loss": 1.2559,
317
+ "step": 500
318
+ },
319
+ {
320
+ "epoch": 1.12,
321
+ "learning_rate": 0.0007984276816171436,
322
+ "loss": 1.3981,
323
+ "step": 510
324
+ },
325
+ {
326
+ "epoch": 1.14,
327
+ "learning_rate": 0.0007984276816171436,
328
+ "loss": 1.3224,
329
+ "step": 520
330
+ },
331
+ {
332
+ "epoch": 1.16,
333
+ "learning_rate": 0.0007984276816171436,
334
+ "loss": 1.3697,
335
+ "step": 530
336
+ },
337
+ {
338
+ "epoch": 1.18,
339
+ "learning_rate": 0.0007984276816171436,
340
+ "loss": 1.2984,
341
+ "step": 540
342
+ },
343
+ {
344
+ "epoch": 1.2,
345
+ "learning_rate": 0.0007984276816171436,
346
+ "loss": 1.2361,
347
+ "step": 550
348
+ },
349
+ {
350
+ "epoch": 1.23,
351
+ "learning_rate": 0.0007984276816171436,
352
+ "loss": 1.2341,
353
+ "step": 560
354
+ },
355
+ {
356
+ "epoch": 1.25,
357
+ "learning_rate": 0.0007984276816171436,
358
+ "loss": 1.2852,
359
+ "step": 570
360
+ },
361
+ {
362
+ "epoch": 1.27,
363
+ "learning_rate": 0.0007984276816171436,
364
+ "loss": 1.2903,
365
+ "step": 580
366
+ },
367
+ {
368
+ "epoch": 1.29,
369
+ "learning_rate": 0.0007984276816171436,
370
+ "loss": 1.3681,
371
+ "step": 590
372
+ },
373
+ {
374
+ "epoch": 1.31,
375
+ "learning_rate": 0.0007984276816171436,
376
+ "loss": 1.2503,
377
+ "step": 600
378
+ },
379
+ {
380
+ "epoch": 1.33,
381
+ "learning_rate": 0.0007984276816171436,
382
+ "loss": 1.3659,
383
+ "step": 610
384
+ },
385
+ {
386
+ "epoch": 1.36,
387
+ "learning_rate": 0.0007984276816171436,
388
+ "loss": 1.2709,
389
+ "step": 620
390
+ },
391
+ {
392
+ "epoch": 1.38,
393
+ "learning_rate": 0.0007984276816171436,
394
+ "loss": 1.261,
395
+ "step": 630
396
+ },
397
+ {
398
+ "epoch": 1.4,
399
+ "learning_rate": 0.0007984276816171436,
400
+ "loss": 1.2984,
401
+ "step": 640
402
+ },
403
+ {
404
+ "epoch": 1.42,
405
+ "learning_rate": 0.0007984276816171436,
406
+ "loss": 1.2224,
407
+ "step": 650
408
+ },
409
+ {
410
+ "epoch": 1.44,
411
+ "learning_rate": 0.0007984276816171436,
412
+ "loss": 1.3484,
413
+ "step": 660
414
+ },
415
+ {
416
+ "epoch": 1.47,
417
+ "learning_rate": 0.0007984276816171436,
418
+ "loss": 1.2559,
419
+ "step": 670
420
+ },
421
+ {
422
+ "epoch": 1.49,
423
+ "learning_rate": 0.0007984276816171436,
424
+ "loss": 1.2761,
425
+ "step": 680
426
+ },
427
+ {
428
+ "epoch": 1.51,
429
+ "learning_rate": 0.0007984276816171436,
430
+ "loss": 1.2682,
431
+ "step": 690
432
+ },
433
+ {
434
+ "epoch": 1.53,
435
+ "learning_rate": 0.0007984276816171436,
436
+ "loss": 1.164,
437
+ "step": 700
438
+ },
439
+ {
440
+ "epoch": 1.55,
441
+ "learning_rate": 0.0007984276816171436,
442
+ "loss": 1.2756,
443
+ "step": 710
444
+ },
445
+ {
446
+ "epoch": 1.58,
447
+ "learning_rate": 0.0007984276816171436,
448
+ "loss": 1.193,
449
+ "step": 720
450
+ },
451
+ {
452
+ "epoch": 1.6,
453
+ "learning_rate": 0.0007984276816171436,
454
+ "loss": 1.176,
455
+ "step": 730
456
+ },
457
+ {
458
+ "epoch": 1.62,
459
+ "learning_rate": 0.0007984276816171436,
460
+ "loss": 1.2107,
461
+ "step": 740
462
+ },
463
+ {
464
+ "epoch": 1.64,
465
+ "learning_rate": 0.0007984276816171436,
466
+ "loss": 1.1533,
467
+ "step": 750
468
+ },
469
+ {
470
+ "epoch": 1.66,
471
+ "learning_rate": 0.0007984276816171436,
472
+ "loss": 1.2375,
473
+ "step": 760
474
+ },
475
+ {
476
+ "epoch": 1.68,
477
+ "learning_rate": 0.0007984276816171436,
478
+ "loss": 1.163,
479
+ "step": 770
480
+ },
481
+ {
482
+ "epoch": 1.71,
483
+ "learning_rate": 0.0007984276816171436,
484
+ "loss": 1.1706,
485
+ "step": 780
486
+ },
487
+ {
488
+ "epoch": 1.73,
489
+ "learning_rate": 0.0007984276816171436,
490
+ "loss": 1.2231,
491
+ "step": 790
492
+ },
493
+ {
494
+ "epoch": 1.75,
495
+ "learning_rate": 0.0007984276816171436,
496
+ "loss": 1.2927,
497
+ "step": 800
498
+ },
499
+ {
500
+ "epoch": 1.77,
501
+ "learning_rate": 0.0007984276816171436,
502
+ "loss": 1.1889,
503
+ "step": 810
504
+ },
505
+ {
506
+ "epoch": 1.79,
507
+ "learning_rate": 0.0007984276816171436,
508
+ "loss": 1.1449,
509
+ "step": 820
510
+ },
511
+ {
512
+ "epoch": 1.82,
513
+ "learning_rate": 0.0007984276816171436,
514
+ "loss": 1.212,
515
+ "step": 830
516
+ },
517
+ {
518
+ "epoch": 1.84,
519
+ "learning_rate": 0.0007984276816171436,
520
+ "loss": 1.1961,
521
+ "step": 840
522
+ },
523
+ {
524
+ "epoch": 1.86,
525
+ "learning_rate": 0.0007984276816171436,
526
+ "loss": 1.1678,
527
+ "step": 850
528
+ },
529
+ {
530
+ "epoch": 1.88,
531
+ "learning_rate": 0.0007984276816171436,
532
+ "loss": 1.1994,
533
+ "step": 860
534
+ },
535
+ {
536
+ "epoch": 1.9,
537
+ "learning_rate": 0.0007984276816171436,
538
+ "loss": 1.1694,
539
+ "step": 870
540
+ },
541
+ {
542
+ "epoch": 1.93,
543
+ "learning_rate": 0.0007984276816171436,
544
+ "loss": 1.0776,
545
+ "step": 880
546
+ },
547
+ {
548
+ "epoch": 1.95,
549
+ "learning_rate": 0.0007984276816171436,
550
+ "loss": 1.1641,
551
+ "step": 890
552
+ },
553
+ {
554
+ "epoch": 1.97,
555
+ "learning_rate": 0.0007984276816171436,
556
+ "loss": 1.0907,
557
+ "step": 900
558
+ },
559
+ {
560
+ "epoch": 1.99,
561
+ "learning_rate": 0.0007984276816171436,
562
+ "loss": 1.2174,
563
+ "step": 910
564
+ },
565
+ {
566
+ "epoch": 2.0,
567
+ "eval_loss": 1.3603260517120361,
568
+ "eval_runtime": 208.2857,
569
+ "eval_samples_per_second": 12.934,
570
+ "eval_steps_per_second": 3.236,
571
+ "step": 914
572
+ }
573
+ ],
574
+ "logging_steps": 10,
575
+ "max_steps": 914,
576
+ "num_train_epochs": 2,
577
+ "save_steps": 500,
578
+ "total_flos": 5.844435036099379e+16,
579
+ "trial_name": null,
580
+ "trial_params": null
581
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06d91a50799d60deccc757ea948871c2f521d38dd063f02306721efc5918cf00
3
+ size 4091