mgh6 commited on
Commit
bd759f5
1 Parent(s): 5165076

Training in progress, step 2000

Browse files
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2ddcdabb355032ab95f0a91327cfea6d659383376a69ee82ecbe9ad20ecd4726
3
  size 135998360
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5e474ea17cc794124e9274b8600ce1124b1831d467c5361e7d435b6783422c2
3
  size 135998360
run-faa58daf/checkpoint-2000/config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "mgh6/esm_cluster_35M_linear",
3
+ "architectures": [
4
+ "EsmForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout": 0.18727005942368125,
8
+ "emb_layer_norm_before": false,
9
+ "esmfold_config": null,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.0,
12
+ "hidden_size": 480,
13
+ "id2label": {
14
+ "0": "LABEL_0"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 1920,
18
+ "is_folding_model": false,
19
+ "label2id": {
20
+ "LABEL_0": 0
21
+ },
22
+ "layer_norm_eps": 1e-05,
23
+ "mask_token_id": 32,
24
+ "max_position_embeddings": 1026,
25
+ "model_type": "esm",
26
+ "num_attention_heads": 20,
27
+ "num_hidden_layers": 12,
28
+ "pad_token_id": 1,
29
+ "position_embedding_type": "rotary",
30
+ "problem_type": "regression",
31
+ "token_dropout": true,
32
+ "torch_dtype": "float32",
33
+ "transformers_version": "4.35.2",
34
+ "use_cache": true,
35
+ "vocab_list": null,
36
+ "vocab_size": 33
37
+ }
run-faa58daf/checkpoint-2000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5e474ea17cc794124e9274b8600ce1124b1831d467c5361e7d435b6783422c2
3
+ size 135998360
run-faa58daf/checkpoint-2000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5dc2ff149bb5bf45644f1df177bd96b1b984fade80db837d9adda44e3fcf2329
3
+ size 268171578
run-faa58daf/checkpoint-2000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac98222cb8423108e113adc1293471b62941ee9b51fcae23ff839a469d0cdd0d
3
+ size 14244
run-faa58daf/checkpoint-2000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f94392d0ab008316b2b9bf665052a399a7b99d12dbf19e79401f8f9833290ef2
3
+ size 1064
run-faa58daf/checkpoint-2000/trainer_state.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.13721677660942078,
3
+ "best_model_checkpoint": "mgh6/TCS_Pair/run-faa58daf/checkpoint-1000",
4
+ "epoch": 4.0230655759688885,
5
+ "eval_steps": 500,
6
+ "global_step": 2000,
7
+ "is_hyper_param_search": true,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 1.01,
13
+ "learning_rate": 0.0006977185696773308,
14
+ "loss": 0.0932,
15
+ "step": 500
16
+ },
17
+ {
18
+ "epoch": 1.01,
19
+ "eval_loss": 0.15397430956363678,
20
+ "eval_runtime": 21.4027,
21
+ "eval_samples_per_second": 223.85,
22
+ "eval_steps_per_second": 27.987,
23
+ "step": 500
24
+ },
25
+ {
26
+ "epoch": 2.01,
27
+ "learning_rate": 0.0006607631369613704,
28
+ "loss": 0.0411,
29
+ "step": 1000
30
+ },
31
+ {
32
+ "epoch": 2.01,
33
+ "eval_loss": 0.13721677660942078,
34
+ "eval_runtime": 20.95,
35
+ "eval_samples_per_second": 228.687,
36
+ "eval_steps_per_second": 28.592,
37
+ "step": 1000
38
+ },
39
+ {
40
+ "epoch": 3.02,
41
+ "learning_rate": 0.0006238077042454101,
42
+ "loss": 0.0246,
43
+ "step": 1500
44
+ },
45
+ {
46
+ "epoch": 3.02,
47
+ "eval_loss": 0.17266754806041718,
48
+ "eval_runtime": 21.3503,
49
+ "eval_samples_per_second": 224.4,
50
+ "eval_steps_per_second": 28.056,
51
+ "step": 1500
52
+ },
53
+ {
54
+ "epoch": 4.02,
55
+ "learning_rate": 0.0005868522715294498,
56
+ "loss": 0.018,
57
+ "step": 2000
58
+ },
59
+ {
60
+ "epoch": 4.02,
61
+ "eval_loss": 0.15012726187705994,
62
+ "eval_runtime": 22.5416,
63
+ "eval_samples_per_second": 212.541,
64
+ "eval_steps_per_second": 26.573,
65
+ "step": 2000
66
+ }
67
+ ],
68
+ "logging_steps": 500,
69
+ "max_steps": 9940,
70
+ "num_train_epochs": 20,
71
+ "save_steps": 500,
72
+ "total_flos": 1.523576375888736e+16,
73
+ "trial_name": null,
74
+ "trial_params": {
75
+ "classifier_dropout": 0.18727005942368125,
76
+ "gradient_accumulation_steps": 15.310000289738827,
77
+ "learning_rate": 0.000734674002393291,
78
+ "weight_decay": 0.06026718993550663
79
+ }
80
+ }
run-faa58daf/checkpoint-2000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:59d46e74d245c1bc18e06275b4a98e13d9ed4372c7020352518f1ee46e02e7c9
3
+ size 4600