-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtrain_v6_150ep_log3.txt
More file actions
175 lines (170 loc) · 16.9 KB
/
Copy pathtrain_v6_150ep_log3.txt
File metadata and controls
175 lines (170 loc) · 16.9 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
====================================================================
v6 ESM-2 (320-dim) + rich (48-dim) -> 368-dim Evoformer TriMul
150 epochs total | 5 warm-up | lr=0.0001 lr_weight=8.0
====================================================================
Loading PDB files from data/pdbs ...
SKIP 1lmb: 'A'
SKIP 2ci2: 'A'
SKIP 6gv0: 'A'
61 training proteins (excluded 7 test proteins)
Split: 55 train / 6 val
Building v6 model from model_v5.pt ...
[train_v6] Shape mismatch — skipping residue_proj.weight ([256, 48] → [256, 368])
[train_v6] Copied 187 tensors from model_v5.pt, skipped 1 (shape mismatch or new keys).
Total parameters: 3,649,800
ΓöÇΓöÇ Phase A: residue_proj warm-up (5 epochs, lr=0.0005) ΓöÇΓöÇ
[esm_utils] ESM-2 (esm2_t6_8M_UR50D, 320-dim) loaded.
Warm-up ep 1/5 train_loss=3.5171 (134s)
Warm-up ep 2/5 train_loss=3.3823 (132s)
Warm-up ep 3/5 train_loss=3.2951 (132s)
Warm-up ep 4/5 train_loss=3.2549 (132s)
Warm-up ep 5/5 train_loss=3.2227 (134s)
ΓöÇΓöÇ Phase B: full fine-tune (145 epochs, lr=0.0001) ΓöÇΓöÇ
Ep 1/145 train=3.1724 val_MSE=39.66 (148s) Γÿà
Ep 2/145 train=3.1552 val_MSE=43.14 (147s)
Ep 3/145 train=3.0919 val_MSE=39.20 (147s) Γÿà
Ep 4/145 train=3.0642 val_MSE=39.11 (140s) Γÿà
Ep 5/145 train=3.0316 val_MSE=41.37 (162s)
Ep 6/145 train=3.0292 val_MSE=41.96 (196s)
Ep 7/145 train=2.9747 val_MSE=40.75 (199s)
Ep 8/145 train=2.9328 val_MSE=40.10 (136s)
Ep 9/145 train=2.9652 val_MSE=39.59 (138s)
Ep 10/145 train=2.9345 val_MSE=39.36 (140s)
Ep 11/145 train=2.9480 val_MSE=39.20 (135s)
Ep 12/145 train=2.9013 val_MSE=39.43 (134s)
Ep 13/145 train=2.8401 val_MSE=39.40 (134s)
Ep 14/145 train=2.8786 val_MSE=38.78 (138s) Γÿà
Ep 15/145 train=2.8671 val_MSE=38.95 (158s)
Ep 16/145 train=2.8491 val_MSE=38.33 (146s) Γÿà
Ep 17/145 train=2.8084 val_MSE=40.14 (148s)
Ep 18/145 train=2.8181 val_MSE=36.66 (136s) Γÿà
Ep 19/145 train=2.8030 val_MSE=38.47 (165s)
Ep 20/145 train=2.7849 val_MSE=37.52 (235s)
Ep 21/145 train=2.7545 val_MSE=37.09 (266s)
Ep 22/145 train=2.7449 val_MSE=38.35 (266s)
Ep 23/145 train=2.7596 val_MSE=37.72 (266s)
Ep 24/145 train=2.7400 val_MSE=38.92 (265s)
Ep 25/145 train=2.7230 val_MSE=39.01 (266s)
Ep 26/145 train=2.7092 val_MSE=37.87 (266s)
Ep 27/145 train=2.7263 val_MSE=36.86 (228s)
Ep 28/145 train=2.6723 val_MSE=38.03 (166s)
Ep 29/145 train=2.6715 val_MSE=37.67 (263s)
Ep 30/145 train=2.6687 val_MSE=38.30 (256s)
Ep 31/145 train=2.6138 val_MSE=39.47 (165s)
Ep 32/145 train=2.6789 val_MSE=37.77 (230s)
Ep 33/145 train=2.6519 val_MSE=37.42 (267s)
Ep 34/145 train=2.6315 val_MSE=38.91 (266s)
Ep 35/145 train=2.6052 val_MSE=38.49 (266s)
Ep 36/145 train=2.5907 val_MSE=37.59 (266s)
Ep 37/145 train=2.5882 val_MSE=37.57 (265s)
Ep 38/145 train=2.5785 val_MSE=37.99 (266s)
Ep 39/145 train=2.5827 val_MSE=38.66 (266s)
Ep 40/145 train=2.5671 val_MSE=38.96 (267s)
Ep 41/145 train=2.6027 val_MSE=37.28 (267s)
Ep 42/145 train=2.5666 val_MSE=38.22 (266s)
Ep 43/145 train=2.5461 val_MSE=40.33 (268s)
Ep 44/145 train=2.5432 val_MSE=38.05 (266s)
Ep 45/145 train=2.5386 val_MSE=38.31 (266s)
Ep 46/145 train=2.5176 val_MSE=39.62 (266s)
Ep 47/145 train=2.5196 val_MSE=39.69 (230s)
Ep 48/145 train=2.4675 val_MSE=39.15 (269s)
Ep 49/145 train=2.4963 val_MSE=38.29 (275s)
Ep 50/145 train=2.4978 val_MSE=38.50 (273s)
Ep 51/145 train=2.5105 val_MSE=38.36 (273s)
Ep 52/145 train=2.4627 val_MSE=40.78 (278s)
Ep 53/145 train=2.4611 val_MSE=39.20 (174s)
Ep 54/145 train=2.4732 val_MSE=39.87 (160s)
Ep 55/145 train=2.4655 val_MSE=40.85 (171s)
Ep 56/145 train=2.4687 val_MSE=41.89 (195s)
Ep 57/145 train=2.4715 val_MSE=39.59 (167s)
Ep 58/145 train=2.4541 val_MSE=39.45 (173s)
Ep 59/145 train=2.4387 val_MSE=41.83 (176s)
Ep 60/145 train=2.4497 val_MSE=40.34 (185s)
Ep 61/145 train=2.4335 val_MSE=39.30 (171s)
Ep 62/145 train=2.4116 val_MSE=39.57 (183s)
Ep 63/145 train=2.4403 val_MSE=40.06 (164s)
Ep 64/145 train=2.3858 val_MSE=40.71 (163s)
Ep 65/145 train=2.4030 val_MSE=40.93 (159s)
Ep 66/145 train=2.3285 val_MSE=40.01 (158s)
Ep 67/145 train=2.3661 val_MSE=39.90 (162s)
Ep 68/145 train=2.3959 val_MSE=40.22 (161s)
Ep 69/145 train=2.3879 val_MSE=39.37 (165s)
Ep 70/145 train=2.3613 val_MSE=40.37 (158s)
Ep 71/145 train=2.3669 val_MSE=39.19 (161s)
Ep 72/145 train=2.3440 val_MSE=41.79 (164s)
Ep 73/145 train=2.3568 val_MSE=40.17 (165s)
Ep 74/145 train=2.3535 val_MSE=39.77 (167s)
Ep 75/145 train=2.2955 val_MSE=39.93 (171s)
Ep 76/145 train=2.3574 val_MSE=41.16 (166s)
Ep 77/145 train=2.3454 val_MSE=39.14 (160s)
Ep 78/145 train=2.3031 val_MSE=39.70 (161s)
Ep 79/145 train=2.3053 val_MSE=40.36 (161s)
Ep 80/145 train=2.3319 val_MSE=40.96 (166s)
Ep 81/145 train=2.3183 val_MSE=41.48 (166s)
Ep 82/145 train=2.3119 val_MSE=40.12 (160s)
Ep 83/145 train=2.3105 val_MSE=40.03 (160s)
Ep 84/145 train=2.2722 val_MSE=40.44 (183s)
Ep 85/145 train=2.2923 val_MSE=40.77 (168s)
Ep 86/145 train=2.2927 val_MSE=39.92 (161s)
Ep 87/145 train=2.2534 val_MSE=40.06 (158s)
Ep 88/145 train=2.2620 val_MSE=39.54 (160s)
Ep 89/145 train=2.2559 val_MSE=39.45 (160s)
Ep 90/145 train=2.3028 val_MSE=40.24 (163s)
Ep 91/145 train=2.2613 val_MSE=40.32 (156s)
Ep 92/145 train=2.2412 val_MSE=40.78 (161s)
Ep 93/145 train=2.2572 val_MSE=40.67 (160s)
Ep 94/145 train=2.2364 val_MSE=39.19 (161s)
Ep 95/145 train=2.2483 val_MSE=40.50 (161s)
Ep 96/145 train=2.2496 val_MSE=41.27 (161s)
Ep 97/145 train=2.2523 val_MSE=41.91 (161s)
Ep 98/145 train=2.2459 val_MSE=39.96 (161s)
Ep 99/145 train=2.2186 val_MSE=41.33 (174s)
Ep 100/145 train=2.2206 val_MSE=40.83 (163s)
Ep 101/145 train=2.1992 val_MSE=40.35 (163s)
Ep 102/145 train=2.1922 val_MSE=40.78 (144s)
Ep 103/145 train=2.2154 val_MSE=40.93 (130s)
Ep 104/145 train=2.2263 val_MSE=41.16 (129s)
Ep 105/145 train=2.2117 val_MSE=40.43 (130s)
Ep 106/145 train=2.2369 val_MSE=40.66 (130s)
Ep 107/145 train=2.1951 val_MSE=40.99 (130s)
Ep 108/145 train=2.1912 val_MSE=41.00 (149s)
Ep 109/145 train=2.1861 val_MSE=40.72 (158s)
Ep 110/145 train=2.1827 val_MSE=40.88 (157s)
Ep 111/145 train=2.2243 val_MSE=40.84 (129s)
Ep 112/145 train=2.1685 val_MSE=41.25 (130s)
Ep 113/145 train=2.1707 val_MSE=41.05 (129s)
Ep 114/145 train=2.1871 val_MSE=40.97 (129s)
Ep 115/145 train=2.1503 val_MSE=40.73 (129s)
Ep 116/145 train=2.1907 val_MSE=41.01 (129s)
Ep 117/145 train=2.1827 val_MSE=41.05 (129s)
Ep 118/145 train=2.2104 val_MSE=40.72 (128s)
Ep 119/145 train=2.1872 val_MSE=40.95 (129s)
Ep 120/145 train=2.1630 val_MSE=40.73 (130s)
Ep 121/145 train=2.1809 val_MSE=40.97 (136s)
Ep 122/145 train=2.1759 val_MSE=40.70 (131s)
Ep 123/145 train=2.1734 val_MSE=41.10 (131s)
Ep 124/145 train=2.1625 val_MSE=41.13 (130s)
Ep 125/145 train=2.2046 val_MSE=41.08 (130s)
Ep 126/145 train=2.1707 val_MSE=41.18 (134s)
Ep 127/145 train=2.1714 val_MSE=40.89 (133s)
Ep 128/145 train=2.1593 val_MSE=40.80 (131s)
Ep 129/145 train=2.1469 val_MSE=41.01 (131s)
Ep 130/145 train=2.2039 val_MSE=40.68 (129s)
Ep 131/145 train=2.1647 val_MSE=40.87 (128s)
Ep 132/145 train=2.1430 val_MSE=40.85 (128s)
Ep 133/145 train=2.1669 val_MSE=40.95 (128s)
Ep 134/145 train=2.1688 val_MSE=40.77 (132s)
Ep 135/145 train=2.1527 val_MSE=41.05 (146s)
Ep 136/145 train=2.1454 val_MSE=41.05 (147s)
Ep 137/145 train=2.1757 val_MSE=40.90 (150s)
Ep 138/145 train=2.1770 val_MSE=40.92 (132s)
Ep 139/145 train=2.1513 val_MSE=40.92 (131s)
Ep 140/145 train=2.1464 val_MSE=40.78 (130s)
Ep 141/145 train=2.1850 val_MSE=40.82 (131s)
Ep 142/145 train=2.1611 val_MSE=40.81 (137s)
Ep 143/145 train=2.1785 val_MSE=40.91 (133s)
Ep 144/145 train=2.1603 val_MSE=40.80 (131s)
Ep 145/145 train=2.1530 val_MSE=40.80 (131s)
Saved: model_v6_150ep.pt (best val_MSE=36.660)
Total training time: 421.7 min
History: model_v6_150ep_history.json