Hugging Face's logo

Promexile
/
scar-weights

Model card Files
xet
 Community
Promexile commited on
Commit
a110495
·
verified ·
1 Parent(s): cab0660

Add selected weights for T2LPP_NPM3D (3 files)

Browse files
Files changed (3) hide show
  1. selected/T2LPP_NPM3D/config/01__ENVIRONMENT.json +44 -0
  2. selected/T2LPP_NPM3D/normalization/NORMALIZATION.json +2250 -0
  3. selected/T2LPP_NPM3D/scar_head/headAB__head_seed0_ht23.pt +3 -0
selected/T2LPP_NPM3D/config/01__ENVIRONMENT.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "package_id": "Text2LocPP-SCAR-HT/NPM3DPose-20-10/bb0/frozen",
3
+ "method": "Text2Loc++ + SCAR(H-T)",
4
+ "dataset_version": "NPM3DPose-20-10",
5
+ "backbone_seed": 0,
6
+ "head_seeds": [
7
+ 0,
8
+ 1,
9
+ 2
10
+ ],
11
+ "captured_at": "2026-10-27T15:15:38",
12
+ "capture_method": "Read-only `python -V` and `pip show torch numpy transformers` on the live machine. `pip show` does not import torch and allocates no GPU.",
13
+ "origin_machine": "E",
14
+ "training_environment": {
15
+ "hostname": "<REMOTE_HOST>-container-p48vc3ykfm-2ab3e0ec",
16
+ "python_executable": "<REMOTE_PATH>",
17
+ "python_env": "conda des4pos",
18
+ "python": "3.10.22",
19
+ "packages": {
20
+ "torch": "2.5.1+cu124",
21
+ "numpy": "2.2.6",
22
+ "transformers": "4.46.3"
23
+ },
24
+ "torch_build_cuda": "cu124",
25
+ "status": "CAPTURED"
26
+ },
27
+ "recovery_environment": null,
28
+ "historical_text2pos_environment": {
29
+ "python_env": "conda t2p",
30
+ "python": "3.10.22",
31
+ "packages": {
32
+ "torch": "1.13.1+cu117",
33
+ "numpy": "1.26.4",
34
+ "transformers": "not installed"
35
+ },
36
+ "torch_build_cuda": "cu117",
37
+ "note": "Text2Pos is historical only; no Text2Pos product is packaged."
38
+ },
39
+ "notes": [
40
+ "torch_build_cuda is the wheel build tag, not the driver or GPU model.",
41
+ "Backup task used no GPU, ran no model and did not touch any training queue.",
42
+ "See code/BACKUP_ENVIRONMENT/ for git commits, uncommitted diffs and package captures."
43
+ ]
44
+ }
selected/T2LPP_NPM3D/normalization/NORMALIZATION.json ADDED
@@ -0,0 +1,2250 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "package_id": "Text2LocPP-SCAR-HT/NPM3DPose-20-10/bb0/frozen",
3
+ "created": "2026-10-27",
4
+ "inference": "(feats - mean) / std per column, then MLP 23->64->32->1 (ReLU, no output activation)",
5
+ "pinned_rule": "columns with TRAIN std < 1e-3 pinned to std=1.0",
6
+ "heads": [
7
+ {
8
+ "method": "Text2Loc++ + SCAR(H-T)",
9
+ "dataset_version": "NPM3DPose-20-10",
10
+ "backbone_seed": 0,
11
+ "head_seed": 0,
12
+ "source_path": "<REMOTE_PATH>",
13
+ "bytes": 27332,
14
+ "mean": [
15
+ 0.3267194926738739,
16
+ 0.5,
17
+ 0.8049779534339905,
18
+ 0.5,
19
+ 0.5,
20
+ 0.5305050611495972,
21
+ 0.5305078625679016,
22
+ 0.0,
23
+ 0.5,
24
+ -0.04762742668390274,
25
+ -0.0031243213452398777,
26
+ 0.2972579300403595,
27
+ 0.0,
28
+ 0.0,
29
+ 0.5,
30
+ 0.5,
31
+ 0.5,
32
+ 0.11817212402820587,
33
+ 0.8049779534339905,
34
+ 0.40905705094337463,
35
+ 0.46698033809661865,
36
+ 1.121683955192566,
37
+ 1.1311274766921997
38
+ ],
39
+ "std": [
40
+ 0.11806745827198029,
41
+ 0.3004623055458069,
42
+ 0.23639965057373047,
43
+ 0.3004623055458069,
44
+ 0.3004623055458069,
45
+ 0.49906668066978455,
46
+ 0.2050330489873886,
47
+ 1.0,
48
+ 0.3004623055458069,
49
+ 0.2283468097448349,
50
+ 0.20042826235294342,
51
+ 0.07892715930938721,
52
+ 1.0,
53
+ 1.0,
54
+ 0.3004623055458069,
55
+ 0.3004623055458069,
56
+ 0.3004623055458069,
57
+ 0.026521125808358192,
58
+ 0.23639965057373047,
59
+ 0.35623958706855774,
60
+ 0.35343223810195923,
61
+ 0.6828075647354126,
62
+ 0.6804105043411255
63
+ ],
64
+ "pinned": [
65
+ false,
66
+ false,
67
+ false,
68
+ false,
69
+ false,
70
+ false,
71
+ false,
72
+ true,
73
+ false,
74
+ false,
75
+ false,
76
+ false,
77
+ true,
78
+ true,
79
+ false,
80
+ false,
81
+ false,
82
+ false,
83
+ false,
84
+ false,
85
+ false,
86
+ false,
87
+ false
88
+ ],
89
+ "feature_names": [
90
+ "cos",
91
+ "cos_rank",
92
+ "sp",
93
+ "p0_rank",
94
+ "p2_rank",
95
+ "is_tied",
96
+ "tied_size",
97
+ "sc_center",
98
+ "sc_center_rank",
99
+ "off_dx",
100
+ "off_dy",
101
+ "off_norm",
102
+ "sc_fine",
103
+ "sat_fine",
104
+ "sc_fine_rank",
105
+ "borda",
106
+ "c_rank",
107
+ "n_obj",
108
+ "n_matched",
109
+ "dist_nearest_matched",
110
+ "dist_centroid_matched",
111
+ "pred_dist_anchor",
112
+ "center_dist_anchor"
113
+ ],
114
+ "config": {
115
+ "epochs": 30,
116
+ "lr": 0.001,
117
+ "wd": 0.0001,
118
+ "batch": 512,
119
+ "hidden": [
120
+ 64,
121
+ 32
122
+ ],
123
+ "loss": "listwise_softmax_ce",
124
+ "seed": 0,
125
+ "k": 25,
126
+ "norms": {
127
+ "pool_norm": 24,
128
+ "rank_norm": 24,
129
+ "tied_norm": 25
130
+ }
131
+ },
132
+ "selection": {
133
+ "rule": "max VAL cell R@1, tie max top1@5m, earlier epoch",
134
+ "epoch": 18,
135
+ "key": [
136
+ 0.2265625,
137
+ 0.1640625
138
+ ],
139
+ "val_cell": {
140
+ "R@1": 0.2265625,
141
+ "R@2": 0.34375,
142
+ "R@3": 0.5234375,
143
+ "R@5": 0.7421875,
144
+ "R@10": 0.921875,
145
+ "R@25": 0.984375
146
+ },
147
+ "val_loc": {
148
+ "top1@5m": 0.1640625,
149
+ "top3@5m": 0.421875,
150
+ "top5@5m": 0.5625,
151
+ "top10@5m": 0.7109375
152
+ }
153
+ },
154
+ "history": [
155
+ {
156
+ "epoch": 1,
157
+ "loss": 3.3257594108581543,
158
+ "train_top1": 0.0,
159
+ "val_cell": {
160
+ "R@1": 0.0078125,
161
+ "R@2": 0.015625,
162
+ "R@3": 0.015625,
163
+ "R@5": 0.015625,
164
+ "R@10": 0.0546875,
165
+ "R@25": 0.984375
166
+ },
167
+ "val_loc": {
168
+ "top1@5m": 0.0,
169
+ "top3@5m": 0.0078125,
170
+ "top5@5m": 0.0078125,
171
+ "top10@5m": 0.046875
172
+ }
173
+ },
174
+ {
175
+ "epoch": 2,
176
+ "loss": 3.3003368377685547,
177
+ "train_top1": 0.020202020202020204,
178
+ "val_cell": {
179
+ "R@1": 0.0234375,
180
+ "R@2": 0.0390625,
181
+ "R@3": 0.0390625,
182
+ "R@5": 0.0546875,
183
+ "R@10": 0.1328125,
184
+ "R@25": 0.984375
185
+ },
186
+ "val_loc": {
187
+ "top1@5m": 0.015625,
188
+ "top3@5m": 0.0234375,
189
+ "top5@5m": 0.0390625,
190
+ "top10@5m": 0.09375
191
+ }
192
+ },
193
+ {
194
+ "epoch": 3,
195
+ "loss": 3.275946617126465,
196
+ "train_top1": 0.030303030303030304,
197
+ "val_cell": {
198
+ "R@1": 0.0546875,
199
+ "R@2": 0.0703125,
200
+ "R@3": 0.0859375,
201
+ "R@5": 0.1328125,
202
+ "R@10": 0.2109375,
203
+ "R@25": 0.984375
204
+ },
205
+ "val_loc": {
206
+ "top1@5m": 0.03125,
207
+ "top3@5m": 0.0546875,
208
+ "top5@5m": 0.09375,
209
+ "top10@5m": 0.1640625
210
+ }
211
+ },
212
+ {
213
+ "epoch": 4,
214
+ "loss": 3.2524728775024414,
215
+ "train_top1": 0.050505050505050504,
216
+ "val_cell": {
217
+ "R@1": 0.078125,
218
+ "R@2": 0.1171875,
219
+ "R@3": 0.1484375,
220
+ "R@5": 0.2109375,
221
+ "R@10": 0.328125,
222
+ "R@25": 0.984375
223
+ },
224
+ "val_loc": {
225
+ "top1@5m": 0.0546875,
226
+ "top3@5m": 0.109375,
227
+ "top5@5m": 0.1640625,
228
+ "top10@5m": 0.2734375
229
+ }
230
+ },
231
+ {
232
+ "epoch": 5,
233
+ "loss": 3.2294976711273193,
234
+ "train_top1": 0.0707070707070707,
235
+ "val_cell": {
236
+ "R@1": 0.0859375,
237
+ "R@2": 0.1328125,
238
+ "R@3": 0.203125,
239
+ "R@5": 0.3125,
240
+ "R@10": 0.4375,
241
+ "R@25": 0.984375
242
+ },
243
+ "val_loc": {
244
+ "top1@5m": 0.0625,
245
+ "top3@5m": 0.15625,
246
+ "top5@5m": 0.25,
247
+ "top10@5m": 0.375
248
+ }
249
+ },
250
+ {
251
+ "epoch": 6,
252
+ "loss": 3.2071900367736816,
253
+ "train_top1": 0.1111111111111111,
254
+ "val_cell": {
255
+ "R@1": 0.0859375,
256
+ "R@2": 0.1796875,
257
+ "R@3": 0.265625,
258
+ "R@5": 0.3984375,
259
+ "R@10": 0.5390625,
260
+ "R@25": 0.984375
261
+ },
262
+ "val_loc": {
263
+ "top1@5m": 0.0703125,
264
+ "top3@5m": 0.21875,
265
+ "top5@5m": 0.3359375,
266
+ "top10@5m": 0.453125
267
+ }
268
+ },
269
+ {
270
+ "epoch": 7,
271
+ "loss": 3.185366153717041,
272
+ "train_top1": 0.13131313131313133,
273
+ "val_cell": {
274
+ "R@1": 0.109375,
275
+ "R@2": 0.2265625,
276
+ "R@3": 0.2890625,
277
+ "R@5": 0.4453125,
278
+ "R@10": 0.703125,
279
+ "R@25": 0.984375
280
+ },
281
+ "val_loc": {
282
+ "top1@5m": 0.0859375,
283
+ "top3@5m": 0.25,
284
+ "top5@5m": 0.3671875,
285
+ "top10@5m": 0.5625
286
+ }
287
+ },
288
+ {
289
+ "epoch": 8,
290
+ "loss": 3.1637918949127197,
291
+ "train_top1": 0.13131313131313133,
292
+ "val_cell": {
293
+ "R@1": 0.109375,
294
+ "R@2": 0.2578125,
295
+ "R@3": 0.328125,
296
+ "R@5": 0.4765625,
297
+ "R@10": 0.7734375,
298
+ "R@25": 0.984375
299
+ },
300
+ "val_loc": {
301
+ "top1@5m": 0.09375,
302
+ "top3@5m": 0.296875,
303
+ "top5@5m": 0.40625,
304
+ "top10@5m": 0.6171875
305
+ }
306
+ },
307
+ {
308
+ "epoch": 9,
309
+ "loss": 3.142493963241577,
310
+ "train_top1": 0.16161616161616163,
311
+ "val_cell": {
312
+ "R@1": 0.1015625,
313
+ "R@2": 0.25,
314
+ "R@3": 0.328125,
315
+ "R@5": 0.53125,
316
+ "R@10": 0.8359375,
317
+ "R@25": 0.984375
318
+ },
319
+ "val_loc": {
320
+ "top1@5m": 0.09375,
321
+ "top3@5m": 0.296875,
322
+ "top5@5m": 0.4296875,
323
+ "top10@5m": 0.6640625
324
+ }
325
+ },
326
+ {
327
+ "epoch": 10,
328
+ "loss": 3.121326446533203,
329
+ "train_top1": 0.16161616161616163,
330
+ "val_cell": {
331
+ "R@1": 0.125,
332
+ "R@2": 0.2734375,
333
+ "R@3": 0.375,
334
+ "R@5": 0.578125,
335
+ "R@10": 0.8984375,
336
+ "R@25": 0.984375
337
+ },
338
+ "val_loc": {
339
+ "top1@5m": 0.1171875,
340
+ "top3@5m": 0.328125,
341
+ "top5@5m": 0.4765625,
342
+ "top10@5m": 0.6953125
343
+ }
344
+ },
345
+ {
346
+ "epoch": 11,
347
+ "loss": 3.100229024887085,
348
+ "train_top1": 0.1717171717171717,
349
+ "val_cell": {
350
+ "R@1": 0.140625,
351
+ "R@2": 0.2890625,
352
+ "R@3": 0.421875,
353
+ "R@5": 0.625,
354
+ "R@10": 0.9140625,
355
+ "R@25": 0.984375
356
+ },
357
+ "val_loc": {
358
+ "top1@5m": 0.1171875,
359
+ "top3@5m": 0.34375,
360
+ "top5@5m": 0.5078125,
361
+ "top10@5m": 0.703125
362
+ }
363
+ },
364
+ {
365
+ "epoch": 12,
366
+ "loss": 3.079233407974243,
367
+ "train_top1": 0.20202020202020202,
368
+ "val_cell": {
369
+ "R@1": 0.125,
370
+ "R@2": 0.3125,
371
+ "R@3": 0.4765625,
372
+ "R@5": 0.6640625,
373
+ "R@10": 0.9140625,
374
+ "R@25": 0.984375
375
+ },
376
+ "val_loc": {
377
+ "top1@5m": 0.109375,
378
+ "top3@5m": 0.3828125,
379
+ "top5@5m": 0.53125,
380
+ "top10@5m": 0.703125
381
+ }
382
+ },
383
+ {
384
+ "epoch": 13,
385
+ "loss": 3.0579841136932373,
386
+ "train_top1": 0.23232323232323232,
387
+ "val_cell": {
388
+ "R@1": 0.140625,
389
+ "R@2": 0.3125,
390
+ "R@3": 0.4765625,
391
+ "R@5": 0.6953125,
392
+ "R@10": 0.9140625,
393
+ "R@25": 0.984375
394
+ },
395
+ "val_loc": {
396
+ "top1@5m": 0.125,
397
+ "top3@5m": 0.390625,
398
+ "top5@5m": 0.5390625,
399
+ "top10@5m": 0.703125
400
+ }
401
+ },
402
+ {
403
+ "epoch": 14,
404
+ "loss": 3.0366299152374268,
405
+ "train_top1": 0.2222222222222222,
406
+ "val_cell": {
407
+ "R@1": 0.1484375,
408
+ "R@2": 0.3203125,
409
+ "R@3": 0.5,
410
+ "R@5": 0.7109375,
411
+ "R@10": 0.921875,
412
+ "R@25": 0.984375
413
+ },
414
+ "val_loc": {
415
+ "top1@5m": 0.1328125,
416
+ "top3@5m": 0.4140625,
417
+ "top5@5m": 0.5625,
418
+ "top10@5m": 0.7109375
419
+ }
420
+ },
421
+ {
422
+ "epoch": 15,
423
+ "loss": 3.015029191970825,
424
+ "train_top1": 0.21212121212121213,
425
+ "val_cell": {
426
+ "R@1": 0.1796875,
427
+ "R@2": 0.328125,
428
+ "R@3": 0.5078125,
429
+ "R@5": 0.71875,
430
+ "R@10": 0.921875,
431
+ "R@25": 0.984375
432
+ },
433
+ "val_loc": {
434
+ "top1@5m": 0.1484375,
435
+ "top3@5m": 0.4140625,
436
+ "top5@5m": 0.5703125,
437
+ "top10@5m": 0.7109375
438
+ }
439
+ },
440
+ {
441
+ "epoch": 16,
442
+ "loss": 2.993229389190674,
443
+ "train_top1": 0.21212121212121213,
444
+ "val_cell": {
445
+ "R@1": 0.203125,
446
+ "R@2": 0.34375,
447
+ "R@3": 0.515625,
448
+ "R@5": 0.7109375,
449
+ "R@10": 0.921875,
450
+ "R@25": 0.984375
451
+ },
452
+ "val_loc": {
453
+ "top1@5m": 0.15625,
454
+ "top3@5m": 0.4140625,
455
+ "top5@5m": 0.5625,
456
+ "top10@5m": 0.7109375
457
+ }
458
+ },
459
+ {
460
+ "epoch": 17,
461
+ "loss": 2.9711060523986816,
462
+ "train_top1": 0.23232323232323232,
463
+ "val_cell": {
464
+ "R@1": 0.21875,
465
+ "R@2": 0.34375,
466
+ "R@3": 0.5390625,
467
+ "R@5": 0.734375,
468
+ "R@10": 0.921875,
469
+ "R@25": 0.984375
470
+ },
471
+ "val_loc": {
472
+ "top1@5m": 0.15625,
473
+ "top3@5m": 0.4296875,
474
+ "top5@5m": 0.5625,
475
+ "top10@5m": 0.7109375
476
+ }
477
+ },
478
+ {
479
+ "epoch": 18,
480
+ "loss": 2.94877552986145,
481
+ "train_top1": 0.2222222222222222,
482
+ "val_cell": {
483
+ "R@1": 0.2265625,
484
+ "R@2": 0.34375,
485
+ "R@3": 0.5234375,
486
+ "R@5": 0.7421875,
487
+ "R@10": 0.921875,
488
+ "R@25": 0.984375
489
+ },
490
+ "val_loc": {
491
+ "top1@5m": 0.1640625,
492
+ "top3@5m": 0.421875,
493
+ "top5@5m": 0.5625,
494
+ "top10@5m": 0.7109375
495
+ }
496
+ },
497
+ {
498
+ "epoch": 19,
499
+ "loss": 2.926208257675171,
500
+ "train_top1": 0.2222222222222222,
501
+ "val_cell": {
502
+ "R@1": 0.2265625,
503
+ "R@2": 0.359375,
504
+ "R@3": 0.5390625,
505
+ "R@5": 0.7578125,
506
+ "R@10": 0.921875,
507
+ "R@25": 0.984375
508
+ },
509
+ "val_loc": {
510
+ "top1@5m": 0.1640625,
511
+ "top3@5m": 0.4375,
512
+ "top5@5m": 0.5703125,
513
+ "top10@5m": 0.7109375
514
+ }
515
+ },
516
+ {
517
+ "epoch": 20,
518
+ "loss": 2.9032318592071533,
519
+ "train_top1": 0.23232323232323232,
520
+ "val_cell": {
521
+ "R@1": 0.2109375,
522
+ "R@2": 0.3515625,
523
+ "R@3": 0.5546875,
524
+ "R@5": 0.7578125,
525
+ "R@10": 0.921875,
526
+ "R@25": 0.984375
527
+ },
528
+ "val_loc": {
529
+ "top1@5m": 0.1484375,
530
+ "top3@5m": 0.4296875,
531
+ "top5@5m": 0.5703125,
532
+ "top10@5m": 0.7109375
533
+ }
534
+ },
535
+ {
536
+ "epoch": 21,
537
+ "loss": 2.880115270614624,
538
+ "train_top1": 0.23232323232323232,
539
+ "val_cell": {
540
+ "R@1": 0.1875,
541
+ "R@2": 0.3515625,
542
+ "R@3": 0.5546875,
543
+ "R@5": 0.7578125,
544
+ "R@10": 0.921875,
545
+ "R@25": 0.984375
546
+ },
547
+ "val_loc": {
548
+ "top1@5m": 0.125,
549
+ "top3@5m": 0.4296875,
550
+ "top5@5m": 0.5703125,
551
+ "top10@5m": 0.7109375
552
+ }
553
+ },
554
+ {
555
+ "epoch": 22,
556
+ "loss": 2.856735944747925,
557
+ "train_top1": 0.2222222222222222,
558
+ "val_cell": {
559
+ "R@1": 0.1875,
560
+ "R@2": 0.359375,
561
+ "R@3": 0.5546875,
562
+ "R@5": 0.7578125,
563
+ "R@10": 0.921875,
564
+ "R@25": 0.984375
565
+ },
566
+ "val_loc": {
567
+ "top1@5m": 0.125,
568
+ "top3@5m": 0.4296875,
569
+ "top5@5m": 0.5703125,
570
+ "top10@5m": 0.7109375
571
+ }
572
+ },
573
+ {
574
+ "epoch": 23,
575
+ "loss": 2.8328795433044434,
576
+ "train_top1": 0.2222222222222222,
577
+ "val_cell": {
578
+ "R@1": 0.171875,
579
+ "R@2": 0.3828125,
580
+ "R@3": 0.546875,
581
+ "R@5": 0.7578125,
582
+ "R@10": 0.921875,
583
+ "R@25": 0.984375
584
+ },
585
+ "val_loc": {
586
+ "top1@5m": 0.109375,
587
+ "top3@5m": 0.421875,
588
+ "top5@5m": 0.5703125,
589
+ "top10@5m": 0.7109375
590
+ }
591
+ },
592
+ {
593
+ "epoch": 24,
594
+ "loss": 2.8085789680480957,
595
+ "train_top1": 0.2222222222222222,
596
+ "val_cell": {
597
+ "R@1": 0.1796875,
598
+ "R@2": 0.390625,
599
+ "R@3": 0.5703125,
600
+ "R@5": 0.765625,
601
+ "R@10": 0.921875,
602
+ "R@25": 0.984375
603
+ },
604
+ "val_loc": {
605
+ "top1@5m": 0.109375,
606
+ "top3@5m": 0.4296875,
607
+ "top5@5m": 0.578125,
608
+ "top10@5m": 0.7109375
609
+ }
610
+ },
611
+ {
612
+ "epoch": 25,
613
+ "loss": 2.7839114665985107,
614
+ "train_top1": 0.2222222222222222,
615
+ "val_cell": {
616
+ "R@1": 0.1875,
617
+ "R@2": 0.390625,
618
+ "R@3": 0.5703125,
619
+ "R@5": 0.765625,
620
+ "R@10": 0.921875,
621
+ "R@25": 0.984375
622
+ },
623
+ "val_loc": {
624
+ "top1@5m": 0.1171875,
625
+ "top3@5m": 0.4296875,
626
+ "top5@5m": 0.578125,
627
+ "top10@5m": 0.7109375
628
+ }
629
+ },
630
+ {
631
+ "epoch": 26,
632
+ "loss": 2.7589504718780518,
633
+ "train_top1": 0.2222222222222222,
634
+ "val_cell": {
635
+ "R@1": 0.1875,
636
+ "R@2": 0.3828125,
637
+ "R@3": 0.5703125,
638
+ "R@5": 0.765625,
639
+ "R@10": 0.921875,
640
+ "R@25": 0.984375
641
+ },
642
+ "val_loc": {
643
+ "top1@5m": 0.1171875,
644
+ "top3@5m": 0.4296875,
645
+ "top5@5m": 0.578125,
646
+ "top10@5m": 0.7109375
647
+ }
648
+ },
649
+ {
650
+ "epoch": 27,
651
+ "loss": 2.7338056564331055,
652
+ "train_top1": 0.2222222222222222,
653
+ "val_cell": {
654
+ "R@1": 0.1875,
655
+ "R@2": 0.3828125,
656
+ "R@3": 0.5703125,
657
+ "R@5": 0.7578125,
658
+ "R@10": 0.921875,
659
+ "R@25": 0.984375
660
+ },
661
+ "val_loc": {
662
+ "top1@5m": 0.1171875,
663
+ "top3@5m": 0.4296875,
664
+ "top5@5m": 0.578125,
665
+ "top10@5m": 0.7109375
666
+ }
667
+ },
668
+ {
669
+ "epoch": 28,
670
+ "loss": 2.708627462387085,
671
+ "train_top1": 0.2222222222222222,
672
+ "val_cell": {
673
+ "R@1": 0.1953125,
674
+ "R@2": 0.375,
675
+ "R@3": 0.578125,
676
+ "R@5": 0.7578125,
677
+ "R@10": 0.921875,
678
+ "R@25": 0.984375
679
+ },
680
+ "val_loc": {
681
+ "top1@5m": 0.125,
682
+ "top3@5m": 0.4375,
683
+ "top5@5m": 0.578125,
684
+ "top10@5m": 0.7109375
685
+ }
686
+ },
687
+ {
688
+ "epoch": 29,
689
+ "loss": 2.6835334300994873,
690
+ "train_top1": 0.2222222222222222,
691
+ "val_cell": {
692
+ "R@1": 0.1953125,
693
+ "R@2": 0.375,
694
+ "R@3": 0.5703125,
695
+ "R@5": 0.75,
696
+ "R@10": 0.9296875,
697
+ "R@25": 0.984375
698
+ },
699
+ "val_loc": {
700
+ "top1@5m": 0.125,
701
+ "top3@5m": 0.4296875,
702
+ "top5@5m": 0.5703125,
703
+ "top10@5m": 0.7109375
704
+ }
705
+ },
706
+ {
707
+ "epoch": 30,
708
+ "loss": 2.6585447788238525,
709
+ "train_top1": 0.23232323232323232,
710
+ "val_cell": {
711
+ "R@1": 0.1953125,
712
+ "R@2": 0.3828125,
713
+ "R@3": 0.5703125,
714
+ "R@5": 0.75,
715
+ "R@10": 0.9296875,
716
+ "R@25": 0.984375
717
+ },
718
+ "val_loc": {
719
+ "top1@5m": 0.125,
720
+ "top3@5m": 0.4296875,
721
+ "top5@5m": 0.5703125,
722
+ "top10@5m": 0.7109375
723
+ }
724
+ }
725
+ ],
726
+ "train_npz": "<REMOTE_PATH>",
727
+ "val_npz": "<REMOTE_PATH>",
728
+ "state_dict_shapes": {
729
+ "net.0.weight": [
730
+ 64,
731
+ 23
732
+ ],
733
+ "net.0.bias": [
734
+ 64
735
+ ],
736
+ "net.2.weight": [
737
+ 32,
738
+ 64
739
+ ],
740
+ "net.2.bias": [
741
+ 32
742
+ ],
743
+ "net.4.weight": [
744
+ 1,
745
+ 32
746
+ ],
747
+ "net.4.bias": [
748
+ 1
749
+ ]
750
+ },
751
+ "file": "heads/head_seed0_ht23.pt",
752
+ "sha256": "3971a3099a48519c1da040b7dc28ef1cbe27ebe00f87d4259b63304452fcb807"
753
+ },
754
+ {
755
+ "method": "Text2Loc++ + SCAR(H-T)",
756
+ "dataset_version": "NPM3DPose-20-10",
757
+ "backbone_seed": 0,
758
+ "head_seed": 1,
759
+ "source_path": "<REMOTE_PATH>",
760
+ "bytes": 27332,
761
+ "mean": [
762
+ 0.3267194926738739,
763
+ 0.5,
764
+ 0.8049779534339905,
765
+ 0.5,
766
+ 0.5,
767
+ 0.5305050611495972,
768
+ 0.5305078625679016,
769
+ 0.0,
770
+ 0.5,
771
+ -0.04762742668390274,
772
+ -0.0031243213452398777,
773
+ 0.2972579300403595,
774
+ 0.0,
775
+ 0.0,
776
+ 0.5,
777
+ 0.5,
778
+ 0.5,
779
+ 0.11817212402820587,
780
+ 0.8049779534339905,
781
+ 0.40905705094337463,
782
+ 0.46698033809661865,
783
+ 1.121683955192566,
784
+ 1.1311274766921997
785
+ ],
786
+ "std": [
787
+ 0.11806745827198029,
788
+ 0.3004623055458069,
789
+ 0.23639965057373047,
790
+ 0.3004623055458069,
791
+ 0.3004623055458069,
792
+ 0.49906668066978455,
793
+ 0.2050330489873886,
794
+ 1.0,
795
+ 0.3004623055458069,
796
+ 0.2283468097448349,
797
+ 0.20042826235294342,
798
+ 0.07892715930938721,
799
+ 1.0,
800
+ 1.0,
801
+ 0.3004623055458069,
802
+ 0.3004623055458069,
803
+ 0.3004623055458069,
804
+ 0.026521125808358192,
805
+ 0.23639965057373047,
806
+ 0.35623958706855774,
807
+ 0.35343223810195923,
808
+ 0.6828075647354126,
809
+ 0.6804105043411255
810
+ ],
811
+ "pinned": [
812
+ false,
813
+ false,
814
+ false,
815
+ false,
816
+ false,
817
+ false,
818
+ false,
819
+ true,
820
+ false,
821
+ false,
822
+ false,
823
+ false,
824
+ true,
825
+ true,
826
+ false,
827
+ false,
828
+ false,
829
+ false,
830
+ false,
831
+ false,
832
+ false,
833
+ false,
834
+ false
835
+ ],
836
+ "feature_names": [
837
+ "cos",
838
+ "cos_rank",
839
+ "sp",
840
+ "p0_rank",
841
+ "p2_rank",
842
+ "is_tied",
843
+ "tied_size",
844
+ "sc_center",
845
+ "sc_center_rank",
846
+ "off_dx",
847
+ "off_dy",
848
+ "off_norm",
849
+ "sc_fine",
850
+ "sat_fine",
851
+ "sc_fine_rank",
852
+ "borda",
853
+ "c_rank",
854
+ "n_obj",
855
+ "n_matched",
856
+ "dist_nearest_matched",
857
+ "dist_centroid_matched",
858
+ "pred_dist_anchor",
859
+ "center_dist_anchor"
860
+ ],
861
+ "config": {
862
+ "epochs": 30,
863
+ "lr": 0.001,
864
+ "wd": 0.0001,
865
+ "batch": 512,
866
+ "hidden": [
867
+ 64,
868
+ 32
869
+ ],
870
+ "loss": "listwise_softmax_ce",
871
+ "seed": 1,
872
+ "k": 25,
873
+ "norms": {
874
+ "pool_norm": 24,
875
+ "rank_norm": 24,
876
+ "tied_norm": 25
877
+ }
878
+ },
879
+ "selection": {
880
+ "rule": "max VAL cell R@1, tie max top1@5m, earlier epoch",
881
+ "epoch": 4,
882
+ "key": [
883
+ 0.203125,
884
+ 0.1328125
885
+ ],
886
+ "val_cell": {
887
+ "R@1": 0.203125,
888
+ "R@2": 0.390625,
889
+ "R@3": 0.4765625,
890
+ "R@5": 0.7109375,
891
+ "R@10": 0.90625,
892
+ "R@25": 0.984375
893
+ },
894
+ "val_loc": {
895
+ "top1@5m": 0.1328125,
896
+ "top3@5m": 0.34375,
897
+ "top5@5m": 0.5234375,
898
+ "top10@5m": 0.6875
899
+ }
900
+ },
901
+ "history": [
902
+ {
903
+ "epoch": 1,
904
+ "loss": 3.1905412673950195,
905
+ "train_top1": 0.1111111111111111,
906
+ "val_cell": {
907
+ "R@1": 0.1640625,
908
+ "R@2": 0.28125,
909
+ "R@3": 0.3984375,
910
+ "R@5": 0.5703125,
911
+ "R@10": 0.78125,
912
+ "R@25": 0.984375
913
+ },
914
+ "val_loc": {
915
+ "top1@5m": 0.0859375,
916
+ "top3@5m": 0.265625,
917
+ "top5@5m": 0.3984375,
918
+ "top10@5m": 0.5859375
919
+ }
920
+ },
921
+ {
922
+ "epoch": 2,
923
+ "loss": 3.162872076034546,
924
+ "train_top1": 0.15151515151515152,
925
+ "val_cell": {
926
+ "R@1": 0.1953125,
927
+ "R@2": 0.34375,
928
+ "R@3": 0.4609375,
929
+ "R@5": 0.640625,
930
+ "R@10": 0.8671875,
931
+ "R@25": 0.984375
932
+ },
933
+ "val_loc": {
934
+ "top1@5m": 0.1171875,
935
+ "top3@5m": 0.3359375,
936
+ "top5@5m": 0.4609375,
937
+ "top10@5m": 0.65625
938
+ }
939
+ },
940
+ {
941
+ "epoch": 3,
942
+ "loss": 3.1363136768341064,
943
+ "train_top1": 0.2222222222222222,
944
+ "val_cell": {
945
+ "R@1": 0.1953125,
946
+ "R@2": 0.3828125,
947
+ "R@3": 0.5,
948
+ "R@5": 0.671875,
949
+ "R@10": 0.90625,
950
+ "R@25": 0.984375
951
+ },
952
+ "val_loc": {
953
+ "top1@5m": 0.125,
954
+ "top3@5m": 0.375,
955
+ "top5@5m": 0.484375,
956
+ "top10@5m": 0.6875
957
+ }
958
+ },
959
+ {
960
+ "epoch": 4,
961
+ "loss": 3.110572576522827,
962
+ "train_top1": 0.23232323232323232,
963
+ "val_cell": {
964
+ "R@1": 0.203125,
965
+ "R@2": 0.390625,
966
+ "R@3": 0.4765625,
967
+ "R@5": 0.7109375,
968
+ "R@10": 0.90625,
969
+ "R@25": 0.984375
970
+ },
971
+ "val_loc": {
972
+ "top1@5m": 0.1328125,
973
+ "top3@5m": 0.34375,
974
+ "top5@5m": 0.5234375,
975
+ "top10@5m": 0.6875
976
+ }
977
+ },
978
+ {
979
+ "epoch": 5,
980
+ "loss": 3.0857646465301514,
981
+ "train_top1": 0.2222222222222222,
982
+ "val_cell": {
983
+ "R@1": 0.1953125,
984
+ "R@2": 0.359375,
985
+ "R@3": 0.4921875,
986
+ "R@5": 0.6953125,
987
+ "R@10": 0.9296875,
988
+ "R@25": 0.984375
989
+ },
990
+ "val_loc": {
991
+ "top1@5m": 0.125,
992
+ "top3@5m": 0.3515625,
993
+ "top5@5m": 0.5078125,
994
+ "top10@5m": 0.7109375
995
+ }
996
+ },
997
+ {
998
+ "epoch": 6,
999
+ "loss": 3.0617878437042236,
1000
+ "train_top1": 0.2222222222222222,
1001
+ "val_cell": {
1002
+ "R@1": 0.1953125,
1003
+ "R@2": 0.375,
1004
+ "R@3": 0.5,
1005
+ "R@5": 0.7109375,
1006
+ "R@10": 0.9296875,
1007
+ "R@25": 0.984375
1008
+ },
1009
+ "val_loc": {
1010
+ "top1@5m": 0.125,
1011
+ "top3@5m": 0.359375,
1012
+ "top5@5m": 0.5234375,
1013
+ "top10@5m": 0.7109375
1014
+ }
1015
+ },
1016
+ {
1017
+ "epoch": 7,
1018
+ "loss": 3.038334608078003,
1019
+ "train_top1": 0.2222222222222222,
1020
+ "val_cell": {
1021
+ "R@1": 0.1953125,
1022
+ "R@2": 0.3671875,
1023
+ "R@3": 0.5,
1024
+ "R@5": 0.71875,
1025
+ "R@10": 0.9296875,
1026
+ "R@25": 0.984375
1027
+ },
1028
+ "val_loc": {
1029
+ "top1@5m": 0.125,
1030
+ "top3@5m": 0.359375,
1031
+ "top5@5m": 0.53125,
1032
+ "top10@5m": 0.7109375
1033
+ }
1034
+ },
1035
+ {
1036
+ "epoch": 8,
1037
+ "loss": 3.0153086185455322,
1038
+ "train_top1": 0.2222222222222222,
1039
+ "val_cell": {
1040
+ "R@1": 0.1953125,
1041
+ "R@2": 0.359375,
1042
+ "R@3": 0.4921875,
1043
+ "R@5": 0.7109375,
1044
+ "R@10": 0.9296875,
1045
+ "R@25": 0.984375
1046
+ },
1047
+ "val_loc": {
1048
+ "top1@5m": 0.125,
1049
+ "top3@5m": 0.3515625,
1050
+ "top5@5m": 0.5234375,
1051
+ "top10@5m": 0.7109375
1052
+ }
1053
+ },
1054
+ {
1055
+ "epoch": 9,
1056
+ "loss": 2.992750644683838,
1057
+ "train_top1": 0.2222222222222222,
1058
+ "val_cell": {
1059
+ "R@1": 0.1953125,
1060
+ "R@2": 0.3671875,
1061
+ "R@3": 0.5,
1062
+ "R@5": 0.734375,
1063
+ "R@10": 0.9296875,
1064
+ "R@25": 0.984375
1065
+ },
1066
+ "val_loc": {
1067
+ "top1@5m": 0.125,
1068
+ "top3@5m": 0.3515625,
1069
+ "top5@5m": 0.5390625,
1070
+ "top10@5m": 0.7109375
1071
+ }
1072
+ },
1073
+ {
1074
+ "epoch": 10,
1075
+ "loss": 2.9704442024230957,
1076
+ "train_top1": 0.2222222222222222,
1077
+ "val_cell": {
1078
+ "R@1": 0.1953125,
1079
+ "R@2": 0.3671875,
1080
+ "R@3": 0.515625,
1081
+ "R@5": 0.734375,
1082
+ "R@10": 0.921875,
1083
+ "R@25": 0.984375
1084
+ },
1085
+ "val_loc": {
1086
+ "top1@5m": 0.125,
1087
+ "top3@5m": 0.3671875,
1088
+ "top5@5m": 0.546875,
1089
+ "top10@5m": 0.703125
1090
+ }
1091
+ },
1092
+ {
1093
+ "epoch": 11,
1094
+ "loss": 2.9482855796813965,
1095
+ "train_top1": 0.2222222222222222,
1096
+ "val_cell": {
1097
+ "R@1": 0.1953125,
1098
+ "R@2": 0.375,
1099
+ "R@3": 0.5234375,
1100
+ "R@5": 0.734375,
1101
+ "R@10": 0.921875,
1102
+ "R@25": 0.984375
1103
+ },
1104
+ "val_loc": {
1105
+ "top1@5m": 0.125,
1106
+ "top3@5m": 0.375,
1107
+ "top5@5m": 0.546875,
1108
+ "top10@5m": 0.703125
1109
+ }
1110
+ },
1111
+ {
1112
+ "epoch": 12,
1113
+ "loss": 2.9262478351593018,
1114
+ "train_top1": 0.2222222222222222,
1115
+ "val_cell": {
1116
+ "R@1": 0.1953125,
1117
+ "R@2": 0.3671875,
1118
+ "R@3": 0.53125,
1119
+ "R@5": 0.734375,
1120
+ "R@10": 0.921875,
1121
+ "R@25": 0.984375
1122
+ },
1123
+ "val_loc": {
1124
+ "top1@5m": 0.125,
1125
+ "top3@5m": 0.3828125,
1126
+ "top5@5m": 0.546875,
1127
+ "top10@5m": 0.703125
1128
+ }
1129
+ },
1130
+ {
1131
+ "epoch": 13,
1132
+ "loss": 2.904334545135498,
1133
+ "train_top1": 0.2222222222222222,
1134
+ "val_cell": {
1135
+ "R@1": 0.1953125,
1136
+ "R@2": 0.359375,
1137
+ "R@3": 0.515625,
1138
+ "R@5": 0.734375,
1139
+ "R@10": 0.921875,
1140
+ "R@25": 0.984375
1141
+ },
1142
+ "val_loc": {
1143
+ "top1@5m": 0.125,
1144
+ "top3@5m": 0.375,
1145
+ "top5@5m": 0.546875,
1146
+ "top10@5m": 0.703125
1147
+ }
1148
+ },
1149
+ {
1150
+ "epoch": 14,
1151
+ "loss": 2.8823964595794678,
1152
+ "train_top1": 0.2222222222222222,
1153
+ "val_cell": {
1154
+ "R@1": 0.1953125,
1155
+ "R@2": 0.359375,
1156
+ "R@3": 0.5078125,
1157
+ "R@5": 0.734375,
1158
+ "R@10": 0.921875,
1159
+ "R@25": 0.984375
1160
+ },
1161
+ "val_loc": {
1162
+ "top1@5m": 0.125,
1163
+ "top3@5m": 0.3671875,
1164
+ "top5@5m": 0.546875,
1165
+ "top10@5m": 0.703125
1166
+ }
1167
+ },
1168
+ {
1169
+ "epoch": 15,
1170
+ "loss": 2.860255479812622,
1171
+ "train_top1": 0.2222222222222222,
1172
+ "val_cell": {
1173
+ "R@1": 0.1953125,
1174
+ "R@2": 0.359375,
1175
+ "R@3": 0.5078125,
1176
+ "R@5": 0.734375,
1177
+ "R@10": 0.921875,
1178
+ "R@25": 0.984375
1179
+ },
1180
+ "val_loc": {
1181
+ "top1@5m": 0.125,
1182
+ "top3@5m": 0.3671875,
1183
+ "top5@5m": 0.546875,
1184
+ "top10@5m": 0.703125
1185
+ }
1186
+ },
1187
+ {
1188
+ "epoch": 16,
1189
+ "loss": 2.8380086421966553,
1190
+ "train_top1": 0.2222222222222222,
1191
+ "val_cell": {
1192
+ "R@1": 0.1953125,
1193
+ "R@2": 0.359375,
1194
+ "R@3": 0.515625,
1195
+ "R@5": 0.734375,
1196
+ "R@10": 0.9296875,
1197
+ "R@25": 0.984375
1198
+ },
1199
+ "val_loc": {
1200
+ "top1@5m": 0.125,
1201
+ "top3@5m": 0.375,
1202
+ "top5@5m": 0.546875,
1203
+ "top10@5m": 0.7109375
1204
+ }
1205
+ },
1206
+ {
1207
+ "epoch": 17,
1208
+ "loss": 2.8157222270965576,
1209
+ "train_top1": 0.2222222222222222,
1210
+ "val_cell": {
1211
+ "R@1": 0.1953125,
1212
+ "R@2": 0.3671875,
1213
+ "R@3": 0.53125,
1214
+ "R@5": 0.734375,
1215
+ "R@10": 0.9296875,
1216
+ "R@25": 0.984375
1217
+ },
1218
+ "val_loc": {
1219
+ "top1@5m": 0.125,
1220
+ "top3@5m": 0.3828125,
1221
+ "top5@5m": 0.546875,
1222
+ "top10@5m": 0.7109375
1223
+ }
1224
+ },
1225
+ {
1226
+ "epoch": 18,
1227
+ "loss": 2.79337215423584,
1228
+ "train_top1": 0.2222222222222222,
1229
+ "val_cell": {
1230
+ "R@1": 0.1953125,
1231
+ "R@2": 0.3671875,
1232
+ "R@3": 0.5234375,
1233
+ "R@5": 0.734375,
1234
+ "R@10": 0.9296875,
1235
+ "R@25": 0.984375
1236
+ },
1237
+ "val_loc": {
1238
+ "top1@5m": 0.125,
1239
+ "top3@5m": 0.375,
1240
+ "top5@5m": 0.546875,
1241
+ "top10@5m": 0.7109375
1242
+ }
1243
+ },
1244
+ {
1245
+ "epoch": 19,
1246
+ "loss": 2.770949125289917,
1247
+ "train_top1": 0.2222222222222222,
1248
+ "val_cell": {
1249
+ "R@1": 0.1953125,
1250
+ "R@2": 0.3671875,
1251
+ "R@3": 0.53125,
1252
+ "R@5": 0.734375,
1253
+ "R@10": 0.9296875,
1254
+ "R@25": 0.984375
1255
+ },
1256
+ "val_loc": {
1257
+ "top1@5m": 0.125,
1258
+ "top3@5m": 0.3828125,
1259
+ "top5@5m": 0.546875,
1260
+ "top10@5m": 0.7109375
1261
+ }
1262
+ },
1263
+ {
1264
+ "epoch": 20,
1265
+ "loss": 2.7485480308532715,
1266
+ "train_top1": 0.2222222222222222,
1267
+ "val_cell": {
1268
+ "R@1": 0.1953125,
1269
+ "R@2": 0.3671875,
1270
+ "R@3": 0.53125,
1271
+ "R@5": 0.734375,
1272
+ "R@10": 0.9296875,
1273
+ "R@25": 0.984375
1274
+ },
1275
+ "val_loc": {
1276
+ "top1@5m": 0.125,
1277
+ "top3@5m": 0.3828125,
1278
+ "top5@5m": 0.546875,
1279
+ "top10@5m": 0.7109375
1280
+ }
1281
+ },
1282
+ {
1283
+ "epoch": 21,
1284
+ "loss": 2.7261807918548584,
1285
+ "train_top1": 0.2222222222222222,
1286
+ "val_cell": {
1287
+ "R@1": 0.1953125,
1288
+ "R@2": 0.3671875,
1289
+ "R@3": 0.5234375,
1290
+ "R@5": 0.734375,
1291
+ "R@10": 0.9296875,
1292
+ "R@25": 0.984375
1293
+ },
1294
+ "val_loc": {
1295
+ "top1@5m": 0.125,
1296
+ "top3@5m": 0.375,
1297
+ "top5@5m": 0.546875,
1298
+ "top10@5m": 0.7109375
1299
+ }
1300
+ },
1301
+ {
1302
+ "epoch": 22,
1303
+ "loss": 2.7038955688476562,
1304
+ "train_top1": 0.2222222222222222,
1305
+ "val_cell": {
1306
+ "R@1": 0.1953125,
1307
+ "R@2": 0.3671875,
1308
+ "R@3": 0.5234375,
1309
+ "R@5": 0.734375,
1310
+ "R@10": 0.9296875,
1311
+ "R@25": 0.984375
1312
+ },
1313
+ "val_loc": {
1314
+ "top1@5m": 0.125,
1315
+ "top3@5m": 0.375,
1316
+ "top5@5m": 0.546875,
1317
+ "top10@5m": 0.7109375
1318
+ }
1319
+ },
1320
+ {
1321
+ "epoch": 23,
1322
+ "loss": 2.6817238330841064,
1323
+ "train_top1": 0.2222222222222222,
1324
+ "val_cell": {
1325
+ "R@1": 0.1953125,
1326
+ "R@2": 0.3671875,
1327
+ "R@3": 0.5390625,
1328
+ "R@5": 0.734375,
1329
+ "R@10": 0.9296875,
1330
+ "R@25": 0.984375
1331
+ },
1332
+ "val_loc": {
1333
+ "top1@5m": 0.125,
1334
+ "top3@5m": 0.390625,
1335
+ "top5@5m": 0.546875,
1336
+ "top10@5m": 0.7109375
1337
+ }
1338
+ },
1339
+ {
1340
+ "epoch": 24,
1341
+ "loss": 2.659811019897461,
1342
+ "train_top1": 0.2222222222222222,
1343
+ "val_cell": {
1344
+ "R@1": 0.1953125,
1345
+ "R@2": 0.3671875,
1346
+ "R@3": 0.53125,
1347
+ "R@5": 0.734375,
1348
+ "R@10": 0.9296875,
1349
+ "R@25": 0.984375
1350
+ },
1351
+ "val_loc": {
1352
+ "top1@5m": 0.125,
1353
+ "top3@5m": 0.3828125,
1354
+ "top5@5m": 0.546875,
1355
+ "top10@5m": 0.7109375
1356
+ }
1357
+ },
1358
+ {
1359
+ "epoch": 25,
1360
+ "loss": 2.6382594108581543,
1361
+ "train_top1": 0.2222222222222222,
1362
+ "val_cell": {
1363
+ "R@1": 0.1953125,
1364
+ "R@2": 0.3671875,
1365
+ "R@3": 0.5390625,
1366
+ "R@5": 0.734375,
1367
+ "R@10": 0.9296875,
1368
+ "R@25": 0.984375
1369
+ },
1370
+ "val_loc": {
1371
+ "top1@5m": 0.125,
1372
+ "top3@5m": 0.390625,
1373
+ "top5@5m": 0.546875,
1374
+ "top10@5m": 0.7109375
1375
+ }
1376
+ },
1377
+ {
1378
+ "epoch": 26,
1379
+ "loss": 2.6171860694885254,
1380
+ "train_top1": 0.2222222222222222,
1381
+ "val_cell": {
1382
+ "R@1": 0.1953125,
1383
+ "R@2": 0.3671875,
1384
+ "R@3": 0.5390625,
1385
+ "R@5": 0.734375,
1386
+ "R@10": 0.9296875,
1387
+ "R@25": 0.984375
1388
+ },
1389
+ "val_loc": {
1390
+ "top1@5m": 0.125,
1391
+ "top3@5m": 0.390625,
1392
+ "top5@5m": 0.546875,
1393
+ "top10@5m": 0.7109375
1394
+ }
1395
+ },
1396
+ {
1397
+ "epoch": 27,
1398
+ "loss": 2.596531391143799,
1399
+ "train_top1": 0.2222222222222222,
1400
+ "val_cell": {
1401
+ "R@1": 0.1953125,
1402
+ "R@2": 0.3671875,
1403
+ "R@3": 0.5390625,
1404
+ "R@5": 0.734375,
1405
+ "R@10": 0.9296875,
1406
+ "R@25": 0.984375
1407
+ },
1408
+ "val_loc": {
1409
+ "top1@5m": 0.125,
1410
+ "top3@5m": 0.390625,
1411
+ "top5@5m": 0.546875,
1412
+ "top10@5m": 0.7109375
1413
+ }
1414
+ },
1415
+ {
1416
+ "epoch": 28,
1417
+ "loss": 2.576479911804199,
1418
+ "train_top1": 0.2222222222222222,
1419
+ "val_cell": {
1420
+ "R@1": 0.1953125,
1421
+ "R@2": 0.3671875,
1422
+ "R@3": 0.5390625,
1423
+ "R@5": 0.7421875,
1424
+ "R@10": 0.9296875,
1425
+ "R@25": 0.984375
1426
+ },
1427
+ "val_loc": {
1428
+ "top1@5m": 0.125,
1429
+ "top3@5m": 0.390625,
1430
+ "top5@5m": 0.5546875,
1431
+ "top10@5m": 0.7109375
1432
+ }
1433
+ },
1434
+ {
1435
+ "epoch": 29,
1436
+ "loss": 2.557147979736328,
1437
+ "train_top1": 0.2222222222222222,
1438
+ "val_cell": {
1439
+ "R@1": 0.1953125,
1440
+ "R@2": 0.3671875,
1441
+ "R@3": 0.5390625,
1442
+ "R@5": 0.75,
1443
+ "R@10": 0.9296875,
1444
+ "R@25": 0.984375
1445
+ },
1446
+ "val_loc": {
1447
+ "top1@5m": 0.125,
1448
+ "top3@5m": 0.390625,
1449
+ "top5@5m": 0.5625,
1450
+ "top10@5m": 0.7109375
1451
+ }
1452
+ },
1453
+ {
1454
+ "epoch": 30,
1455
+ "loss": 2.5386569499969482,
1456
+ "train_top1": 0.2222222222222222,
1457
+ "val_cell": {
1458
+ "R@1": 0.1953125,
1459
+ "R@2": 0.3671875,
1460
+ "R@3": 0.5390625,
1461
+ "R@5": 0.75,
1462
+ "R@10": 0.9296875,
1463
+ "R@25": 0.984375
1464
+ },
1465
+ "val_loc": {
1466
+ "top1@5m": 0.125,
1467
+ "top3@5m": 0.390625,
1468
+ "top5@5m": 0.5625,
1469
+ "top10@5m": 0.7109375
1470
+ }
1471
+ }
1472
+ ],
1473
+ "train_npz": "<REMOTE_PATH>",
1474
+ "val_npz": "<REMOTE_PATH>",
1475
+ "state_dict_shapes": {
1476
+ "net.0.weight": [
1477
+ 64,
1478
+ 23
1479
+ ],
1480
+ "net.0.bias": [
1481
+ 64
1482
+ ],
1483
+ "net.2.weight": [
1484
+ 32,
1485
+ 64
1486
+ ],
1487
+ "net.2.bias": [
1488
+ 32
1489
+ ],
1490
+ "net.4.weight": [
1491
+ 1,
1492
+ 32
1493
+ ],
1494
+ "net.4.bias": [
1495
+ 1
1496
+ ]
1497
+ },
1498
+ "file": "heads/head_seed1_ht23.pt",
1499
+ "sha256": "9c2a36aef65c8207c68a60829e19341a8a66655401dfb61bcb033979af1c01b3"
1500
+ },
1501
+ {
1502
+ "method": "Text2Loc++ + SCAR(H-T)",
1503
+ "dataset_version": "NPM3DPose-20-10",
1504
+ "backbone_seed": 0,
1505
+ "head_seed": 2,
1506
+ "source_path": "<REMOTE_PATH>",
1507
+ "bytes": 27332,
1508
+ "mean": [
1509
+ 0.3267194926738739,
1510
+ 0.5,
1511
+ 0.8049779534339905,
1512
+ 0.5,
1513
+ 0.5,
1514
+ 0.5305050611495972,
1515
+ 0.5305078625679016,
1516
+ 0.0,
1517
+ 0.5,
1518
+ -0.04762742668390274,
1519
+ -0.0031243213452398777,
1520
+ 0.2972579300403595,
1521
+ 0.0,
1522
+ 0.0,
1523
+ 0.5,
1524
+ 0.5,
1525
+ 0.5,
1526
+ 0.11817212402820587,
1527
+ 0.8049779534339905,
1528
+ 0.40905705094337463,
1529
+ 0.46698033809661865,
1530
+ 1.121683955192566,
1531
+ 1.1311274766921997
1532
+ ],
1533
+ "std": [
1534
+ 0.11806745827198029,
1535
+ 0.3004623055458069,
1536
+ 0.23639965057373047,
1537
+ 0.3004623055458069,
1538
+ 0.3004623055458069,
1539
+ 0.49906668066978455,
1540
+ 0.2050330489873886,
1541
+ 1.0,
1542
+ 0.3004623055458069,
1543
+ 0.2283468097448349,
1544
+ 0.20042826235294342,
1545
+ 0.07892715930938721,
1546
+ 1.0,
1547
+ 1.0,
1548
+ 0.3004623055458069,
1549
+ 0.3004623055458069,
1550
+ 0.3004623055458069,
1551
+ 0.026521125808358192,
1552
+ 0.23639965057373047,
1553
+ 0.35623958706855774,
1554
+ 0.35343223810195923,
1555
+ 0.6828075647354126,
1556
+ 0.6804105043411255
1557
+ ],
1558
+ "pinned": [
1559
+ false,
1560
+ false,
1561
+ false,
1562
+ false,
1563
+ false,
1564
+ false,
1565
+ false,
1566
+ true,
1567
+ false,
1568
+ false,
1569
+ false,
1570
+ false,
1571
+ true,
1572
+ true,
1573
+ false,
1574
+ false,
1575
+ false,
1576
+ false,
1577
+ false,
1578
+ false,
1579
+ false,
1580
+ false,
1581
+ false
1582
+ ],
1583
+ "feature_names": [
1584
+ "cos",
1585
+ "cos_rank",
1586
+ "sp",
1587
+ "p0_rank",
1588
+ "p2_rank",
1589
+ "is_tied",
1590
+ "tied_size",
1591
+ "sc_center",
1592
+ "sc_center_rank",
1593
+ "off_dx",
1594
+ "off_dy",
1595
+ "off_norm",
1596
+ "sc_fine",
1597
+ "sat_fine",
1598
+ "sc_fine_rank",
1599
+ "borda",
1600
+ "c_rank",
1601
+ "n_obj",
1602
+ "n_matched",
1603
+ "dist_nearest_matched",
1604
+ "dist_centroid_matched",
1605
+ "pred_dist_anchor",
1606
+ "center_dist_anchor"
1607
+ ],
1608
+ "config": {
1609
+ "epochs": 30,
1610
+ "lr": 0.001,
1611
+ "wd": 0.0001,
1612
+ "batch": 512,
1613
+ "hidden": [
1614
+ 64,
1615
+ 32
1616
+ ],
1617
+ "loss": "listwise_softmax_ce",
1618
+ "seed": 2,
1619
+ "k": 25,
1620
+ "norms": {
1621
+ "pool_norm": 24,
1622
+ "rank_norm": 24,
1623
+ "tied_norm": 25
1624
+ }
1625
+ },
1626
+ "selection": {
1627
+ "rule": "max VAL cell R@1, tie max top1@5m, earlier epoch",
1628
+ "epoch": 5,
1629
+ "key": [
1630
+ 0.21875,
1631
+ 0.140625
1632
+ ],
1633
+ "val_cell": {
1634
+ "R@1": 0.21875,
1635
+ "R@2": 0.359375,
1636
+ "R@3": 0.515625,
1637
+ "R@5": 0.734375,
1638
+ "R@10": 0.90625,
1639
+ "R@25": 0.984375
1640
+ },
1641
+ "val_loc": {
1642
+ "top1@5m": 0.140625,
1643
+ "top3@5m": 0.359375,
1644
+ "top5@5m": 0.5546875,
1645
+ "top10@5m": 0.6953125
1646
+ }
1647
+ },
1648
+ "history": [
1649
+ {
1650
+ "epoch": 1,
1651
+ "loss": 3.2514865398406982,
1652
+ "train_top1": 0.030303030303030304,
1653
+ "val_cell": {
1654
+ "R@1": 0.046875,
1655
+ "R@2": 0.078125,
1656
+ "R@3": 0.1015625,
1657
+ "R@5": 0.1171875,
1658
+ "R@10": 0.2734375,
1659
+ "R@25": 0.984375
1660
+ },
1661
+ "val_loc": {
1662
+ "top1@5m": 0.0390625,
1663
+ "top3@5m": 0.0703125,
1664
+ "top5@5m": 0.078125,
1665
+ "top10@5m": 0.203125
1666
+ }
1667
+ },
1668
+ {
1669
+ "epoch": 2,
1670
+ "loss": 3.227739095687866,
1671
+ "train_top1": 0.1111111111111111,
1672
+ "val_cell": {
1673
+ "R@1": 0.0859375,
1674
+ "R@2": 0.1953125,
1675
+ "R@3": 0.2578125,
1676
+ "R@5": 0.3984375,
1677
+ "R@10": 0.609375,
1678
+ "R@25": 0.984375
1679
+ },
1680
+ "val_loc": {
1681
+ "top1@5m": 0.0546875,
1682
+ "top3@5m": 0.171875,
1683
+ "top5@5m": 0.296875,
1684
+ "top10@5m": 0.4765625
1685
+ }
1686
+ },
1687
+ {
1688
+ "epoch": 3,
1689
+ "loss": 3.20416259765625,
1690
+ "train_top1": 0.1414141414141414,
1691
+ "val_cell": {
1692
+ "R@1": 0.140625,
1693
+ "R@2": 0.3203125,
1694
+ "R@3": 0.4765625,
1695
+ "R@5": 0.625,
1696
+ "R@10": 0.84375,
1697
+ "R@25": 0.984375
1698
+ },
1699
+ "val_loc": {
1700
+ "top1@5m": 0.09375,
1701
+ "top3@5m": 0.328125,
1702
+ "top5@5m": 0.453125,
1703
+ "top10@5m": 0.6328125
1704
+ }
1705
+ },
1706
+ {
1707
+ "epoch": 4,
1708
+ "loss": 3.180680513381958,
1709
+ "train_top1": 0.20202020202020202,
1710
+ "val_cell": {
1711
+ "R@1": 0.203125,
1712
+ "R@2": 0.3671875,
1713
+ "R@3": 0.5078125,
1714
+ "R@5": 0.6875,
1715
+ "R@10": 0.875,
1716
+ "R@25": 0.984375
1717
+ },
1718
+ "val_loc": {
1719
+ "top1@5m": 0.125,
1720
+ "top3@5m": 0.359375,
1721
+ "top5@5m": 0.5078125,
1722
+ "top10@5m": 0.6640625
1723
+ }
1724
+ },
1725
+ {
1726
+ "epoch": 5,
1727
+ "loss": 3.157688617706299,
1728
+ "train_top1": 0.2222222222222222,
1729
+ "val_cell": {
1730
+ "R@1": 0.21875,
1731
+ "R@2": 0.359375,
1732
+ "R@3": 0.515625,
1733
+ "R@5": 0.734375,
1734
+ "R@10": 0.90625,
1735
+ "R@25": 0.984375
1736
+ },
1737
+ "val_loc": {
1738
+ "top1@5m": 0.140625,
1739
+ "top3@5m": 0.359375,
1740
+ "top5@5m": 0.5546875,
1741
+ "top10@5m": 0.6953125
1742
+ }
1743
+ },
1744
+ {
1745
+ "epoch": 6,
1746
+ "loss": 3.135166645050049,
1747
+ "train_top1": 0.23232323232323232,
1748
+ "val_cell": {
1749
+ "R@1": 0.203125,
1750
+ "R@2": 0.375,
1751
+ "R@3": 0.5390625,
1752
+ "R@5": 0.7421875,
1753
+ "R@10": 0.9140625,
1754
+ "R@25": 0.984375
1755
+ },
1756
+ "val_loc": {
1757
+ "top1@5m": 0.125,
1758
+ "top3@5m": 0.390625,
1759
+ "top5@5m": 0.5625,
1760
+ "top10@5m": 0.703125
1761
+ }
1762
+ },
1763
+ {
1764
+ "epoch": 7,
1765
+ "loss": 3.1129212379455566,
1766
+ "train_top1": 0.2222222222222222,
1767
+ "val_cell": {
1768
+ "R@1": 0.203125,
1769
+ "R@2": 0.3828125,
1770
+ "R@3": 0.5625,
1771
+ "R@5": 0.75,
1772
+ "R@10": 0.9140625,
1773
+ "R@25": 0.984375
1774
+ },
1775
+ "val_loc": {
1776
+ "top1@5m": 0.125,
1777
+ "top3@5m": 0.4140625,
1778
+ "top5@5m": 0.5703125,
1779
+ "top10@5m": 0.703125
1780
+ }
1781
+ },
1782
+ {
1783
+ "epoch": 8,
1784
+ "loss": 3.0905544757843018,
1785
+ "train_top1": 0.2222222222222222,
1786
+ "val_cell": {
1787
+ "R@1": 0.203125,
1788
+ "R@2": 0.375,
1789
+ "R@3": 0.5703125,
1790
+ "R@5": 0.765625,
1791
+ "R@10": 0.9140625,
1792
+ "R@25": 0.984375
1793
+ },
1794
+ "val_loc": {
1795
+ "top1@5m": 0.125,
1796
+ "top3@5m": 0.4140625,
1797
+ "top5@5m": 0.5859375,
1798
+ "top10@5m": 0.703125
1799
+ }
1800
+ },
1801
+ {
1802
+ "epoch": 9,
1803
+ "loss": 3.0680043697357178,
1804
+ "train_top1": 0.2222222222222222,
1805
+ "val_cell": {
1806
+ "R@1": 0.203125,
1807
+ "R@2": 0.390625,
1808
+ "R@3": 0.5703125,
1809
+ "R@5": 0.765625,
1810
+ "R@10": 0.921875,
1811
+ "R@25": 0.984375
1812
+ },
1813
+ "val_loc": {
1814
+ "top1@5m": 0.125,
1815
+ "top3@5m": 0.4140625,
1816
+ "top5@5m": 0.578125,
1817
+ "top10@5m": 0.7109375
1818
+ }
1819
+ },
1820
+ {
1821
+ "epoch": 10,
1822
+ "loss": 3.045168161392212,
1823
+ "train_top1": 0.2222222222222222,
1824
+ "val_cell": {
1825
+ "R@1": 0.203125,
1826
+ "R@2": 0.390625,
1827
+ "R@3": 0.5703125,
1828
+ "R@5": 0.7734375,
1829
+ "R@10": 0.921875,
1830
+ "R@25": 0.984375
1831
+ },
1832
+ "val_loc": {
1833
+ "top1@5m": 0.125,
1834
+ "top3@5m": 0.4140625,
1835
+ "top5@5m": 0.5859375,
1836
+ "top10@5m": 0.7109375
1837
+ }
1838
+ },
1839
+ {
1840
+ "epoch": 11,
1841
+ "loss": 3.0220787525177,
1842
+ "train_top1": 0.2222222222222222,
1843
+ "val_cell": {
1844
+ "R@1": 0.203125,
1845
+ "R@2": 0.390625,
1846
+ "R@3": 0.5703125,
1847
+ "R@5": 0.7734375,
1848
+ "R@10": 0.921875,
1849
+ "R@25": 0.984375
1850
+ },
1851
+ "val_loc": {
1852
+ "top1@5m": 0.125,
1853
+ "top3@5m": 0.4140625,
1854
+ "top5@5m": 0.5859375,
1855
+ "top10@5m": 0.7109375
1856
+ }
1857
+ },
1858
+ {
1859
+ "epoch": 12,
1860
+ "loss": 2.998765707015991,
1861
+ "train_top1": 0.21212121212121213,
1862
+ "val_cell": {
1863
+ "R@1": 0.203125,
1864
+ "R@2": 0.390625,
1865
+ "R@3": 0.5703125,
1866
+ "R@5": 0.7734375,
1867
+ "R@10": 0.921875,
1868
+ "R@25": 0.984375
1869
+ },
1870
+ "val_loc": {
1871
+ "top1@5m": 0.125,
1872
+ "top3@5m": 0.4140625,
1873
+ "top5@5m": 0.5859375,
1874
+ "top10@5m": 0.7109375
1875
+ }
1876
+ },
1877
+ {
1878
+ "epoch": 13,
1879
+ "loss": 2.97517991065979,
1880
+ "train_top1": 0.20202020202020202,
1881
+ "val_cell": {
1882
+ "R@1": 0.203125,
1883
+ "R@2": 0.3828125,
1884
+ "R@3": 0.578125,
1885
+ "R@5": 0.7734375,
1886
+ "R@10": 0.921875,
1887
+ "R@25": 0.984375
1888
+ },
1889
+ "val_loc": {
1890
+ "top1@5m": 0.125,
1891
+ "top3@5m": 0.421875,
1892
+ "top5@5m": 0.5859375,
1893
+ "top10@5m": 0.7109375
1894
+ }
1895
+ },
1896
+ {
1897
+ "epoch": 14,
1898
+ "loss": 2.9512240886688232,
1899
+ "train_top1": 0.21212121212121213,
1900
+ "val_cell": {
1901
+ "R@1": 0.203125,
1902
+ "R@2": 0.390625,
1903
+ "R@3": 0.578125,
1904
+ "R@5": 0.7734375,
1905
+ "R@10": 0.921875,
1906
+ "R@25": 0.984375
1907
+ },
1908
+ "val_loc": {
1909
+ "top1@5m": 0.125,
1910
+ "top3@5m": 0.421875,
1911
+ "top5@5m": 0.5859375,
1912
+ "top10@5m": 0.7109375
1913
+ }
1914
+ },
1915
+ {
1916
+ "epoch": 15,
1917
+ "loss": 2.9270243644714355,
1918
+ "train_top1": 0.21212121212121213,
1919
+ "val_cell": {
1920
+ "R@1": 0.203125,
1921
+ "R@2": 0.3984375,
1922
+ "R@3": 0.5703125,
1923
+ "R@5": 0.7734375,
1924
+ "R@10": 0.921875,
1925
+ "R@25": 0.984375
1926
+ },
1927
+ "val_loc": {
1928
+ "top1@5m": 0.125,
1929
+ "top3@5m": 0.4140625,
1930
+ "top5@5m": 0.5859375,
1931
+ "top10@5m": 0.7109375
1932
+ }
1933
+ },
1934
+ {
1935
+ "epoch": 16,
1936
+ "loss": 2.9024713039398193,
1937
+ "train_top1": 0.21212121212121213,
1938
+ "val_cell": {
1939
+ "R@1": 0.203125,
1940
+ "R@2": 0.3984375,
1941
+ "R@3": 0.5703125,
1942
+ "R@5": 0.765625,
1943
+ "R@10": 0.921875,
1944
+ "R@25": 0.984375
1945
+ },
1946
+ "val_loc": {
1947
+ "top1@5m": 0.125,
1948
+ "top3@5m": 0.4140625,
1949
+ "top5@5m": 0.578125,
1950
+ "top10@5m": 0.7109375
1951
+ }
1952
+ },
1953
+ {
1954
+ "epoch": 17,
1955
+ "loss": 2.8775668144226074,
1956
+ "train_top1": 0.21212121212121213,
1957
+ "val_cell": {
1958
+ "R@1": 0.203125,
1959
+ "R@2": 0.3984375,
1960
+ "R@3": 0.5703125,
1961
+ "R@5": 0.765625,
1962
+ "R@10": 0.921875,
1963
+ "R@25": 0.984375
1964
+ },
1965
+ "val_loc": {
1966
+ "top1@5m": 0.125,
1967
+ "top3@5m": 0.4140625,
1968
+ "top5@5m": 0.578125,
1969
+ "top10@5m": 0.7109375
1970
+ }
1971
+ },
1972
+ {
1973
+ "epoch": 18,
1974
+ "loss": 2.8523294925689697,
1975
+ "train_top1": 0.21212121212121213,
1976
+ "val_cell": {
1977
+ "R@1": 0.203125,
1978
+ "R@2": 0.3984375,
1979
+ "R@3": 0.578125,
1980
+ "R@5": 0.765625,
1981
+ "R@10": 0.921875,
1982
+ "R@25": 0.984375
1983
+ },
1984
+ "val_loc": {
1985
+ "top1@5m": 0.125,
1986
+ "top3@5m": 0.421875,
1987
+ "top5@5m": 0.578125,
1988
+ "top10@5m": 0.7109375
1989
+ }
1990
+ },
1991
+ {
1992
+ "epoch": 19,
1993
+ "loss": 2.826690912246704,
1994
+ "train_top1": 0.21212121212121213,
1995
+ "val_cell": {
1996
+ "R@1": 0.203125,
1997
+ "R@2": 0.3984375,
1998
+ "R@3": 0.5546875,
1999
+ "R@5": 0.765625,
2000
+ "R@10": 0.921875,
2001
+ "R@25": 0.984375
2002
+ },
2003
+ "val_loc": {
2004
+ "top1@5m": 0.125,
2005
+ "top3@5m": 0.4140625,
2006
+ "top5@5m": 0.578125,
2007
+ "top10@5m": 0.7109375
2008
+ }
2009
+ },
2010
+ {
2011
+ "epoch": 20,
2012
+ "loss": 2.8006794452667236,
2013
+ "train_top1": 0.21212121212121213,
2014
+ "val_cell": {
2015
+ "R@1": 0.2109375,
2016
+ "R@2": 0.3984375,
2017
+ "R@3": 0.546875,
2018
+ "R@5": 0.765625,
2019
+ "R@10": 0.921875,
2020
+ "R@25": 0.984375
2021
+ },
2022
+ "val_loc": {
2023
+ "top1@5m": 0.1328125,
2024
+ "top3@5m": 0.40625,
2025
+ "top5@5m": 0.578125,
2026
+ "top10@5m": 0.7109375
2027
+ }
2028
+ },
2029
+ {
2030
+ "epoch": 21,
2031
+ "loss": 2.77433443069458,
2032
+ "train_top1": 0.2222222222222222,
2033
+ "val_cell": {
2034
+ "R@1": 0.2109375,
2035
+ "R@2": 0.3828125,
2036
+ "R@3": 0.546875,
2037
+ "R@5": 0.765625,
2038
+ "R@10": 0.921875,
2039
+ "R@25": 0.984375
2040
+ },
2041
+ "val_loc": {
2042
+ "top1@5m": 0.1328125,
2043
+ "top3@5m": 0.40625,
2044
+ "top5@5m": 0.578125,
2045
+ "top10@5m": 0.7109375
2046
+ }
2047
+ },
2048
+ {
2049
+ "epoch": 22,
2050
+ "loss": 2.7476041316986084,
2051
+ "train_top1": 0.2222222222222222,
2052
+ "val_cell": {
2053
+ "R@1": 0.21875,
2054
+ "R@2": 0.375,
2055
+ "R@3": 0.546875,
2056
+ "R@5": 0.7578125,
2057
+ "R@10": 0.9296875,
2058
+ "R@25": 0.984375
2059
+ },
2060
+ "val_loc": {
2061
+ "top1@5m": 0.140625,
2062
+ "top3@5m": 0.40625,
2063
+ "top5@5m": 0.5703125,
2064
+ "top10@5m": 0.7109375
2065
+ }
2066
+ },
2067
+ {
2068
+ "epoch": 23,
2069
+ "loss": 2.72047758102417,
2070
+ "train_top1": 0.23232323232323232,
2071
+ "val_cell": {
2072
+ "R@1": 0.21875,
2073
+ "R@2": 0.375,
2074
+ "R@3": 0.546875,
2075
+ "R@5": 0.7578125,
2076
+ "R@10": 0.9296875,
2077
+ "R@25": 0.984375
2078
+ },
2079
+ "val_loc": {
2080
+ "top1@5m": 0.140625,
2081
+ "top3@5m": 0.40625,
2082
+ "top5@5m": 0.5703125,
2083
+ "top10@5m": 0.7109375
2084
+ }
2085
+ },
2086
+ {
2087
+ "epoch": 24,
2088
+ "loss": 2.6932363510131836,
2089
+ "train_top1": 0.2222222222222222,
2090
+ "val_cell": {
2091
+ "R@1": 0.21875,
2092
+ "R@2": 0.375,
2093
+ "R@3": 0.546875,
2094
+ "R@5": 0.75,
2095
+ "R@10": 0.9296875,
2096
+ "R@25": 0.984375
2097
+ },
2098
+ "val_loc": {
2099
+ "top1@5m": 0.140625,
2100
+ "top3@5m": 0.40625,
2101
+ "top5@5m": 0.5703125,
2102
+ "top10@5m": 0.7109375
2103
+ }
2104
+ },
2105
+ {
2106
+ "epoch": 25,
2107
+ "loss": 2.6661784648895264,
2108
+ "train_top1": 0.2222222222222222,
2109
+ "val_cell": {
2110
+ "R@1": 0.2109375,
2111
+ "R@2": 0.3828125,
2112
+ "R@3": 0.546875,
2113
+ "R@5": 0.75,
2114
+ "R@10": 0.9296875,
2115
+ "R@25": 0.984375
2116
+ },
2117
+ "val_loc": {
2118
+ "top1@5m": 0.1328125,
2119
+ "top3@5m": 0.40625,
2120
+ "top5@5m": 0.5703125,
2121
+ "top10@5m": 0.7109375
2122
+ }
2123
+ },
2124
+ {
2125
+ "epoch": 26,
2126
+ "loss": 2.6394498348236084,
2127
+ "train_top1": 0.2222222222222222,
2128
+ "val_cell": {
2129
+ "R@1": 0.2109375,
2130
+ "R@2": 0.3828125,
2131
+ "R@3": 0.5625,
2132
+ "R@5": 0.75,
2133
+ "R@10": 0.9296875,
2134
+ "R@25": 0.984375
2135
+ },
2136
+ "val_loc": {
2137
+ "top1@5m": 0.1328125,
2138
+ "top3@5m": 0.4140625,
2139
+ "top5@5m": 0.5703125,
2140
+ "top10@5m": 0.7109375
2141
+ }
2142
+ },
2143
+ {
2144
+ "epoch": 27,
2145
+ "loss": 2.613234758377075,
2146
+ "train_top1": 0.2222222222222222,
2147
+ "val_cell": {
2148
+ "R@1": 0.203125,
2149
+ "R@2": 0.375,
2150
+ "R@3": 0.5703125,
2151
+ "R@5": 0.75,
2152
+ "R@10": 0.9296875,
2153
+ "R@25": 0.984375
2154
+ },
2155
+ "val_loc": {
2156
+ "top1@5m": 0.1328125,
2157
+ "top3@5m": 0.421875,
2158
+ "top5@5m": 0.5703125,
2159
+ "top10@5m": 0.7109375
2160
+ }
2161
+ },
2162
+ {
2163
+ "epoch": 28,
2164
+ "loss": 2.587695598602295,
2165
+ "train_top1": 0.2222222222222222,
2166
+ "val_cell": {
2167
+ "R@1": 0.1953125,
2168
+ "R@2": 0.375,
2169
+ "R@3": 0.5703125,
2170
+ "R@5": 0.75,
2171
+ "R@10": 0.9296875,
2172
+ "R@25": 0.984375
2173
+ },
2174
+ "val_loc": {
2175
+ "top1@5m": 0.125,
2176
+ "top3@5m": 0.421875,
2177
+ "top5@5m": 0.5703125,
2178
+ "top10@5m": 0.7109375
2179
+ }
2180
+ },
2181
+ {
2182
+ "epoch": 29,
2183
+ "loss": 2.5630574226379395,
2184
+ "train_top1": 0.2222222222222222,
2185
+ "val_cell": {
2186
+ "R@1": 0.1953125,
2187
+ "R@2": 0.375,
2188
+ "R@3": 0.5703125,
2189
+ "R@5": 0.75,
2190
+ "R@10": 0.9296875,
2191
+ "R@25": 0.984375
2192
+ },
2193
+ "val_loc": {
2194
+ "top1@5m": 0.125,
2195
+ "top3@5m": 0.421875,
2196
+ "top5@5m": 0.5703125,
2197
+ "top10@5m": 0.7109375
2198
+ }
2199
+ },
2200
+ {
2201
+ "epoch": 30,
2202
+ "loss": 2.5395126342773438,
2203
+ "train_top1": 0.2222222222222222,
2204
+ "val_cell": {
2205
+ "R@1": 0.1953125,
2206
+ "R@2": 0.375,
2207
+ "R@3": 0.5703125,
2208
+ "R@5": 0.75,
2209
+ "R@10": 0.9296875,
2210
+ "R@25": 0.984375
2211
+ },
2212
+ "val_loc": {
2213
+ "top1@5m": 0.125,
2214
+ "top3@5m": 0.421875,
2215
+ "top5@5m": 0.5703125,
2216
+ "top10@5m": 0.7109375
2217
+ }
2218
+ }
2219
+ ],
2220
+ "train_npz": "<REMOTE_PATH>",
2221
+ "val_npz": "<REMOTE_PATH>",
2222
+ "state_dict_shapes": {
2223
+ "net.0.weight": [
2224
+ 64,
2225
+ 23
2226
+ ],
2227
+ "net.0.bias": [
2228
+ 64
2229
+ ],
2230
+ "net.2.weight": [
2231
+ 32,
2232
+ 64
2233
+ ],
2234
+ "net.2.bias": [
2235
+ 32
2236
+ ],
2237
+ "net.4.weight": [
2238
+ 1,
2239
+ 32
2240
+ ],
2241
+ "net.4.bias": [
2242
+ 1
2243
+ ]
2244
+ },
2245
+ "file": "heads/head_seed2_ht23.pt",
2246
+ "sha256": "37aede1481428b48605907ce27158042da5ba0bac27f1c2b7b93b4db706caa6d"
2247
+ }
2248
+ ],
2249
+ "ablation_heads": []
2250
+ }
selected/T2LPP_NPM3D/scar_head/headAB__head_seed0_ht23.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3971a3099a48519c1da040b7dc28ef1cbe27ebe00f87d4259b63304452fcb807
3
+ size 27332