MengHsir commited on
Commit
5dab54d
·
verified ·
1 Parent(s): 15cf61b

Upload folder using huggingface_hub

Browse files
Files changed (23) hide show
  1. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695.ckpt +3 -0
  2. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__A100-SXM4-40GB__la_0.5.json +71 -0
  3. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__A100-SXM4-40GB__la_0.5__relation_results.json +263 -0
  4. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.1.json +71 -0
  5. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.1__relation_results.json +263 -0
  6. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.2.json +71 -0
  7. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.2__relation_results.json +263 -0
  8. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.3.json +71 -0
  9. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.3__relation_results.json +263 -0
  10. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__no__la.json +71 -0
  11. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__no__la__relation_results.json +263 -0
  12. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/last.ckpt +3 -0
  13. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/config.json +376 -0
  14. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1753832749.hnode003.3197765.0 +3 -0
  15. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1753891515.hnode003.3956668.0 +3 -0
  16. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1753892532.hnode003.4069729.0 +3 -0
  17. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1753973183.hnode003.1067497.0 +3 -0
  18. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1753973834.hnode003.1095230.0 +3 -0
  19. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1754129498.hnode003.3387473.0 +3 -0
  20. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1755449537.hnode003.2175788.0 +3 -0
  21. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1755449544.hnode003.2168874.0 +3 -0
  22. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/hparams.yaml +1 -0
  23. ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/training_args.yaml +58 -0
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a2b66e8689fac215dc0eae1a363ee3cbe849c544c2ad5b77c96c5cf95e14126
3
+ size 593077348
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__A100-SXM4-40GB__la_0.5.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "AP50": 0.31085976942345733,
3
+ "(single)R@20": 0.19190484572331626,
4
+ "(single)R@50": 0.23232076332940768,
5
+ "(single)R@100": 0.2567892828655026,
6
+ "(single)mR@20": 0.1203792385316211,
7
+ "(single)mR@50": 0.1838395314543699,
8
+ "(single)mR@100": 0.2274595694865592,
9
+ "eval_arg": {
10
+ "data_path": "dataset/visual_genome",
11
+ "output_path": "./output_2",
12
+ "backbone_dirpath": "",
13
+ "architecture": "SenseTime/deformable-detr",
14
+ "auxiliary_loss": false,
15
+ "from_scratch": false,
16
+ "pretrained": "./ckpt/pretrained_detr__SenseTime__deformable-detr/batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune/version_0/",
17
+ "num_queries": 200,
18
+ "ce_loss_coefficient": 2.0,
19
+ "rel_loss_coefficient": 15.0,
20
+ "connectivity_loss_coefficient": 30.0,
21
+ "smoothing": 1e-14,
22
+ "rel_sample_negatives": 80,
23
+ "rel_sample_nonmatching": 80,
24
+ "rel_sample_negatives_largest": true,
25
+ "rel_sample_nonmatching_largest": true,
26
+ "batch_size": 12,
27
+ "accumulate": 2,
28
+ "gpus": 1,
29
+ "max_epochs": 70,
30
+ "max_epochs_finetune": 25,
31
+ "lr_backbone": 1e-07,
32
+ "lr": 1e-06,
33
+ "lr_initialized": 0.0001,
34
+ "weight_decay": 0.0001,
35
+ "gradient_clip_val": 0.1,
36
+ "frozen_params": false,
37
+ "debug": false,
38
+ "resume": true,
39
+ "memo": "07-27-egtr-3-NO6-fintune",
40
+ "version": 0,
41
+ "patience": 15,
42
+ "finetune": true,
43
+ "skip_main_train": true,
44
+ "main_trained_path": "output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/",
45
+ "filter_duplicate_rels": true,
46
+ "filter_multiple_rels": true,
47
+ "use_freq_bias": true,
48
+ "use_log_softmax": false,
49
+ "skip_train": true,
50
+ "split": "test",
51
+ "eval_batch_size": 3,
52
+ "eval_when_train_end": true,
53
+ "eval_single_preds": true,
54
+ "eval_multiple_preds": false,
55
+ "logit_adjustment": true,
56
+ "logit_adj_tau": 0.5,
57
+ "num_workers": 8,
58
+ "precision": 32,
59
+ "depth_data_path": "dataset/vg_depth",
60
+ "depth_file_suffix": "_depth.pt",
61
+ "use_depth_input": true,
62
+ "depth_strength_factor": 1.0,
63
+ "enable_correlation": true,
64
+ "use_depth_cross_attention": true,
65
+ "depth_fusion_heads": 8,
66
+ "depth_fusion_method": "concat",
67
+ "depth_decoder_fusion_layers": 6
68
+ },
69
+ "weight_source": "finetune",
70
+ "weight_path": "./output_2/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695.ckpt"
71
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__A100-SXM4-40GB__la_0.5__relation_results.json ADDED
@@ -0,0 +1,263 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "sgdet_recall_with_constraint_R@K": {
3
+ "above": {
4
+ "R@20": 0.09375121147509206,
5
+ "R@50": 0.136057052400336,
6
+ "R@100": 0.15856949021128125
7
+ },
8
+ "across": {
9
+ "R@20": 0.047619047619047616,
10
+ "R@50": 0.1111111111111111,
11
+ "R@100": 0.1746031746031746
12
+ },
13
+ "against": {
14
+ "R@20": 0.08870967741935484,
15
+ "R@50": 0.1532258064516129,
16
+ "R@100": 0.20967741935483872
17
+ },
18
+ "along": {
19
+ "R@20": 0.0481651376146789,
20
+ "R@50": 0.06498470948012233,
21
+ "R@100": 0.13149847094801223
22
+ },
23
+ "and": {
24
+ "R@20": 0.09566074950690334,
25
+ "R@50": 0.17751479289940827,
26
+ "R@100": 0.25345167652859957
27
+ },
28
+ "at": {
29
+ "R@20": 0.2489746263468891,
30
+ "R@50": 0.31777314332058854,
31
+ "R@100": 0.35130923415594945
32
+ },
33
+ "attached to": {
34
+ "R@20": 0.027496069182389938,
35
+ "R@50": 0.041374663072776284,
36
+ "R@100": 0.07065925426774483
37
+ },
38
+ "behind": {
39
+ "R@20": 0.13236341786254222,
40
+ "R@50": 0.21238878101446082,
41
+ "R@100": 0.2595480843651182
42
+ },
43
+ "belonging to": {
44
+ "R@20": 0.06081519329926973,
45
+ "R@50": 0.14077278878234292,
46
+ "R@100": 0.21267527589183638
47
+ },
48
+ "between": {
49
+ "R@20": 0.03263888888888889,
50
+ "R@50": 0.07824074074074074,
51
+ "R@100": 0.12685185185185185
52
+ },
53
+ "carrying": {
54
+ "R@20": 0.24841085271317828,
55
+ "R@50": 0.30333333333333334,
56
+ "R@100": 0.34112403100775196
57
+ },
58
+ "covered in": {
59
+ "R@20": 0.08095238095238094,
60
+ "R@50": 0.14166666666666666,
61
+ "R@100": 0.20952380952380953
62
+ },
63
+ "covering": {
64
+ "R@20": 0.06344086021505375,
65
+ "R@50": 0.13413978494623655,
66
+ "R@100": 0.1986559139784946
67
+ },
68
+ "eating": {
69
+ "R@20": 0.10596026490066225,
70
+ "R@50": 0.1479028697571744,
71
+ "R@100": 0.22130242825607063
72
+ },
73
+ "flying in": {
74
+ "R@20": 0.0,
75
+ "R@50": 0.4090909090909091,
76
+ "R@100": 0.49242424242424243
77
+ },
78
+ "for": {
79
+ "R@20": 0.09508196721311475,
80
+ "R@50": 0.128551912568306,
81
+ "R@100": 0.1662568306010929
82
+ },
83
+ "from": {
84
+ "R@20": 0.014084507042253521,
85
+ "R@50": 0.028169014084507043,
86
+ "R@100": 0.04225352112676056
87
+ },
88
+ "growing on": {
89
+ "R@20": 0.01293103448275862,
90
+ "R@50": 0.10775862068965517,
91
+ "R@100": 0.14655172413793102
92
+ },
93
+ "hanging from": {
94
+ "R@20": 0.125,
95
+ "R@50": 0.15139442231075698,
96
+ "R@100": 0.16766268260292164
97
+ },
98
+ "has": {
99
+ "R@20": 0.27637132297722705,
100
+ "R@50": 0.34619057477572235,
101
+ "R@100": 0.38561314217531345
102
+ },
103
+ "holding": {
104
+ "R@20": 0.2359710455415977,
105
+ "R@50": 0.2777535949621839,
106
+ "R@100": 0.3002825656506638
107
+ },
108
+ "in": {
109
+ "R@20": 0.09015441387880412,
110
+ "R@50": 0.11939243601926527,
111
+ "R@100": 0.1382950518316372
112
+ },
113
+ "in front of": {
114
+ "R@20": 0.103192867105001,
115
+ "R@50": 0.15623879258816495,
116
+ "R@100": 0.19832386929667264
117
+ },
118
+ "laying on": {
119
+ "R@20": 0.23048048048048048,
120
+ "R@50": 0.26576576576576577,
121
+ "R@100": 0.30067567567567566
122
+ },
123
+ "looking at": {
124
+ "R@20": 0.0536813922356091,
125
+ "R@50": 0.08514056224899598,
126
+ "R@100": 0.09819277108433734
127
+ },
128
+ "lying on": {
129
+ "R@20": 0.1292517006802721,
130
+ "R@50": 0.1989795918367347,
131
+ "R@100": 0.22448979591836735
132
+ },
133
+ "made of": {
134
+ "R@20": 0.0,
135
+ "R@50": 0.0,
136
+ "R@100": 0.0
137
+ },
138
+ "mounted on": {
139
+ "R@20": 0.11493055555555555,
140
+ "R@50": 0.20069444444444443,
141
+ "R@100": 0.26215277777777773
142
+ },
143
+ "near": {
144
+ "R@20": 0.08224208743759794,
145
+ "R@50": 0.11013279235581913,
146
+ "R@100": 0.13233815692179196
147
+ },
148
+ "of": {
149
+ "R@20": 0.19956552551257672,
150
+ "R@50": 0.25437480720519895,
151
+ "R@100": 0.2724176359346398
152
+ },
153
+ "on": {
154
+ "R@20": 0.13177983480233005,
155
+ "R@50": 0.15392783433725135,
156
+ "R@100": 0.1637968615306533
157
+ },
158
+ "on back of": {
159
+ "R@20": 0.18912529550827423,
160
+ "R@50": 0.2541371158392435,
161
+ "R@100": 0.29314420803782504
162
+ },
163
+ "over": {
164
+ "R@20": 0.09826302729528537,
165
+ "R@50": 0.18622828784119108,
166
+ "R@100": 0.24152191894127376
167
+ },
168
+ "painted on": {
169
+ "R@20": 0.11982758620689656,
170
+ "R@50": 0.1721264367816092,
171
+ "R@100": 0.18764367816091954
172
+ },
173
+ "parked on": {
174
+ "R@20": 0.26787051899411446,
175
+ "R@50": 0.3740502942750133,
176
+ "R@100": 0.46647940074906363
177
+ },
178
+ "part of": {
179
+ "R@20": 0.02821869488536155,
180
+ "R@50": 0.13109935332157555,
181
+ "R@100": 0.20291005291005287
182
+ },
183
+ "playing": {
184
+ "R@20": 0.09090909090909091,
185
+ "R@50": 0.09090909090909091,
186
+ "R@100": 0.18181818181818182
187
+ },
188
+ "riding": {
189
+ "R@20": 0.409092706850182,
190
+ "R@50": 0.4680377577387544,
191
+ "R@100": 0.5079602383589095
192
+ },
193
+ "says": {
194
+ "R@20": 0.0,
195
+ "R@50": 0.08333333333333333,
196
+ "R@100": 0.08333333333333333
197
+ },
198
+ "sitting on": {
199
+ "R@20": 0.1346372767857143,
200
+ "R@50": 0.19252325148809524,
201
+ "R@100": 0.23345424107142856
202
+ },
203
+ "standing on": {
204
+ "R@20": 0.09198961499493413,
205
+ "R@50": 0.11742126815265111,
206
+ "R@100": 0.13745672914555893
207
+ },
208
+ "to": {
209
+ "R@20": 0.2015027322404372,
210
+ "R@50": 0.29849726775956287,
211
+ "R@100": 0.32991803278688525
212
+ },
213
+ "under": {
214
+ "R@20": 0.16433418150975404,
215
+ "R@50": 0.23539722929035903,
216
+ "R@100": 0.27489397794741305
217
+ },
218
+ "using": {
219
+ "R@20": 0.20229166666666662,
220
+ "R@50": 0.29916666666666664,
221
+ "R@100": 0.36604166666666665
222
+ },
223
+ "walking in": {
224
+ "R@20": 0.10563380281690139,
225
+ "R@50": 0.244131455399061,
226
+ "R@100": 0.27511737089201876
227
+ },
228
+ "walking on": {
229
+ "R@20": 0.14405719392314567,
230
+ "R@50": 0.20685880250223412,
231
+ "R@100": 0.24054959785522786
232
+ },
233
+ "watching": {
234
+ "R@20": 0.05779220779220779,
235
+ "R@50": 0.11629389129389128,
236
+ "R@100": 0.15980439313772646
237
+ },
238
+ "wearing": {
239
+ "R@20": 0.42432847697583415,
240
+ "R@50": 0.47827642619859706,
241
+ "R@100": 0.504113118885478
242
+ },
243
+ "wears": {
244
+ "R@20": 0.004271476032273374,
245
+ "R@50": 0.052344397942404586,
246
+ "R@100": 0.20103796627883005
247
+ },
248
+ "with": {
249
+ "R@20": 0.0151392632524708,
250
+ "R@50": 0.03710192672456824,
251
+ "R@100": 0.07460294771615525
252
+ }
253
+ },
254
+ "sgdet_recall_with_constraint_mR@K": {
255
+ "mR@20": 0.1203792385316211,
256
+ "mR@50": 0.1838395314543699,
257
+ "mR@100": 0.2274595694865592,
258
+ "valid_relations": 50,
259
+ "total_relations": 50
260
+ },
261
+ "sgdet_recall_without_constraint_R@K": {},
262
+ "sgdet_recall_without_constraint_mR@K": {}
263
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.1.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "AP50": 0.3108353009300483,
3
+ "(single)R@20": 0.23464301820624542,
4
+ "(single)R@50": 0.30025732086366447,
5
+ "(single)R@100": 0.34345918947420007,
6
+ "(single)mR@20": 0.0685558184377165,
7
+ "(single)mR@50": 0.10355626064595619,
8
+ "(single)mR@100": 0.13606674084998174,
9
+ "eval_arg": {
10
+ "data_path": "dataset/visual_genome",
11
+ "output_path": "./output",
12
+ "backbone_dirpath": "",
13
+ "architecture": "SenseTime/deformable-detr",
14
+ "auxiliary_loss": false,
15
+ "from_scratch": false,
16
+ "pretrained": "./ckpt/pretrained_detr__SenseTime__deformable-detr/batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune/version_0/",
17
+ "num_queries": 200,
18
+ "ce_loss_coefficient": 2.0,
19
+ "rel_loss_coefficient": 15.0,
20
+ "connectivity_loss_coefficient": 30.0,
21
+ "smoothing": 1e-14,
22
+ "rel_sample_negatives": 80,
23
+ "rel_sample_nonmatching": 80,
24
+ "rel_sample_negatives_largest": true,
25
+ "rel_sample_nonmatching_largest": true,
26
+ "batch_size": 12,
27
+ "accumulate": 2,
28
+ "gpus": 1,
29
+ "max_epochs": 70,
30
+ "max_epochs_finetune": 25,
31
+ "lr_backbone": 1e-07,
32
+ "lr": 1e-06,
33
+ "lr_initialized": 0.0001,
34
+ "weight_decay": 0.0001,
35
+ "gradient_clip_val": 0.1,
36
+ "frozen_params": false,
37
+ "debug": false,
38
+ "resume": true,
39
+ "memo": "07-27-egtr-3-NO6-fintune",
40
+ "version": 0,
41
+ "patience": 15,
42
+ "finetune": true,
43
+ "skip_main_train": true,
44
+ "main_trained_path": "output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/",
45
+ "filter_duplicate_rels": true,
46
+ "filter_multiple_rels": true,
47
+ "use_freq_bias": true,
48
+ "use_log_softmax": false,
49
+ "skip_train": true,
50
+ "split": "test",
51
+ "eval_batch_size": 3,
52
+ "eval_when_train_end": true,
53
+ "eval_single_preds": true,
54
+ "eval_multiple_preds": false,
55
+ "logit_adjustment": true,
56
+ "logit_adj_tau": 0.1,
57
+ "num_workers": 8,
58
+ "precision": 32,
59
+ "depth_data_path": "dataset/vg_depth",
60
+ "depth_file_suffix": "_depth.pt",
61
+ "use_depth_input": true,
62
+ "depth_strength_factor": 1.0,
63
+ "enable_correlation": true,
64
+ "use_depth_cross_attention": true,
65
+ "depth_fusion_heads": 8,
66
+ "depth_fusion_method": "concat",
67
+ "depth_decoder_fusion_layers": 6
68
+ },
69
+ "weight_source": "finetune",
70
+ "weight_path": "./output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695.ckpt"
71
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.1__relation_results.json ADDED
@@ -0,0 +1,263 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "sgdet_recall_with_constraint_R@K": {
3
+ "above": {
4
+ "R@20": 0.07562189054726368,
5
+ "R@50": 0.10920398009950247,
6
+ "R@100": 0.125318052594172
7
+ },
8
+ "across": {
9
+ "R@20": 0.007936507936507936,
10
+ "R@50": 0.023809523809523808,
11
+ "R@100": 0.023809523809523808
12
+ },
13
+ "against": {
14
+ "R@20": 0.024193548387096774,
15
+ "R@50": 0.04032258064516129,
16
+ "R@100": 0.04032258064516129
17
+ },
18
+ "along": {
19
+ "R@20": 0.009174311926605505,
20
+ "R@50": 0.01834862385321101,
21
+ "R@100": 0.039755351681957186
22
+ },
23
+ "and": {
24
+ "R@20": 0.01775147928994083,
25
+ "R@50": 0.06015779092702169,
26
+ "R@100": 0.10207100591715976
27
+ },
28
+ "at": {
29
+ "R@20": 0.15444907890163367,
30
+ "R@50": 0.2406036380488935,
31
+ "R@100": 0.3006198586490557
32
+ },
33
+ "attached to": {
34
+ "R@20": 0.0026533018867924527,
35
+ "R@50": 0.0072720125786163515,
36
+ "R@100": 0.015723270440251572
37
+ },
38
+ "behind": {
39
+ "R@20": 0.1303902713359806,
40
+ "R@50": 0.22684879268988406,
41
+ "R@100": 0.29268662247923005
42
+ },
43
+ "belonging to": {
44
+ "R@20": 0.0,
45
+ "R@50": 0.0007961783439490446,
46
+ "R@100": 0.002502274795268426
47
+ },
48
+ "between": {
49
+ "R@20": 0.001388888888888889,
50
+ "R@50": 0.004861111111111111,
51
+ "R@100": 0.018750000000000003
52
+ },
53
+ "carrying": {
54
+ "R@20": 0.14360465116279067,
55
+ "R@50": 0.2051937984496124,
56
+ "R@100": 0.23511627906976743
57
+ },
58
+ "covered in": {
59
+ "R@20": 0.014285714285714285,
60
+ "R@50": 0.07202380952380952,
61
+ "R@100": 0.1113095238095238
62
+ },
63
+ "covering": {
64
+ "R@20": 0.012903225806451613,
65
+ "R@50": 0.04731182795698925,
66
+ "R@100": 0.1010752688172043
67
+ },
68
+ "eating": {
69
+ "R@20": 0.0728476821192053,
70
+ "R@50": 0.09381898454746136,
71
+ "R@100": 0.11589403973509933
72
+ },
73
+ "flying in": {
74
+ "R@20": 0.0,
75
+ "R@50": 0.0,
76
+ "R@100": 0.06060606060606061
77
+ },
78
+ "for": {
79
+ "R@20": 0.004098360655737705,
80
+ "R@50": 0.018442622950819672,
81
+ "R@100": 0.03210382513661202
82
+ },
83
+ "from": {
84
+ "R@20": 0.0,
85
+ "R@50": 0.014084507042253521,
86
+ "R@100": 0.014084507042253521
87
+ },
88
+ "growing on": {
89
+ "R@20": 0.0,
90
+ "R@50": 0.0,
91
+ "R@100": 0.004310344827586207
92
+ },
93
+ "hanging from": {
94
+ "R@20": 0.04731075697211155,
95
+ "R@50": 0.07569721115537849,
96
+ "R@100": 0.08964143426294821
97
+ },
98
+ "has": {
99
+ "R@20": 0.2993442119380327,
100
+ "R@50": 0.3809683819526443,
101
+ "R@100": 0.43174050582696827
102
+ },
103
+ "holding": {
104
+ "R@20": 0.26921689875677607,
105
+ "R@50": 0.3360220079851982,
106
+ "R@100": 0.37786980231765505
107
+ },
108
+ "in": {
109
+ "R@20": 0.09800878749817056,
110
+ "R@50": 0.13656938029290971,
111
+ "R@100": 0.16687650812442778
112
+ },
113
+ "in front of": {
114
+ "R@20": 0.07154064554692169,
115
+ "R@50": 0.10757621040047817,
116
+ "R@100": 0.15479677226539149
117
+ },
118
+ "laying on": {
119
+ "R@20": 0.04504504504504504,
120
+ "R@50": 0.09271771771771772,
121
+ "R@100": 0.13025525525525528
122
+ },
123
+ "looking at": {
124
+ "R@20": 0.018741633199464522,
125
+ "R@50": 0.05488621151271753,
126
+ "R@100": 0.07295850066934405
127
+ },
128
+ "lying on": {
129
+ "R@20": 0.04081632653061224,
130
+ "R@50": 0.061224489795918366,
131
+ "R@100": 0.08163265306122448
132
+ },
133
+ "made of": {
134
+ "R@20": 0.0,
135
+ "R@50": 0.0,
136
+ "R@100": 0.0
137
+ },
138
+ "mounted on": {
139
+ "R@20": 0.0,
140
+ "R@50": 0.0,
141
+ "R@100": 0.0
142
+ },
143
+ "near": {
144
+ "R@20": 0.12523888065162503,
145
+ "R@50": 0.19232593512101118,
146
+ "R@100": 0.2438225310238344
147
+ },
148
+ "of": {
149
+ "R@20": 0.21399538823129008,
150
+ "R@50": 0.3111187585131015,
151
+ "R@100": 0.3642678316996516
152
+ },
153
+ "on": {
154
+ "R@20": 0.24538279812576158,
155
+ "R@50": 0.31321311201124696,
156
+ "R@100": 0.3581433952934044
157
+ },
158
+ "on back of": {
159
+ "R@20": 0.07092198581560284,
160
+ "R@50": 0.09456264775413711,
161
+ "R@100": 0.16312056737588654
162
+ },
163
+ "over": {
164
+ "R@20": 0.04813895781637717,
165
+ "R@50": 0.12514474772539289,
166
+ "R@100": 0.16699751861042184
167
+ },
168
+ "painted on": {
169
+ "R@20": 0.0,
170
+ "R@50": 0.020689655172413793,
171
+ "R@100": 0.03793103448275863
172
+ },
173
+ "parked on": {
174
+ "R@20": 0.0,
175
+ "R@50": 0.011235955056179775,
176
+ "R@100": 0.04325842696629214
177
+ },
178
+ "part of": {
179
+ "R@20": 0.0,
180
+ "R@50": 0.0,
181
+ "R@100": 0.003527336860670194
182
+ },
183
+ "playing": {
184
+ "R@20": 0.0,
185
+ "R@50": 0.045454545454545456,
186
+ "R@100": 0.045454545454545456
187
+ },
188
+ "riding": {
189
+ "R@20": 0.24399488477561568,
190
+ "R@50": 0.31519538047777257,
191
+ "R@100": 0.3738332542319253
192
+ },
193
+ "says": {
194
+ "R@20": 0.0,
195
+ "R@50": 0.0,
196
+ "R@100": 0.08333333333333333
197
+ },
198
+ "sitting on": {
199
+ "R@20": 0.09591889880952381,
200
+ "R@50": 0.14075613839285714,
201
+ "R@100": 0.18025669642857142
202
+ },
203
+ "standing on": {
204
+ "R@20": 0.016828140830800406,
205
+ "R@50": 0.0450491810199257,
206
+ "R@100": 0.08712955927051672
207
+ },
208
+ "to": {
209
+ "R@20": 0.0,
210
+ "R@50": 0.04918032786885246,
211
+ "R@100": 0.13934426229508196
212
+ },
213
+ "under": {
214
+ "R@20": 0.15213457732541702,
215
+ "R@50": 0.22404580152671755,
216
+ "R@100": 0.2750494769578739
217
+ },
218
+ "using": {
219
+ "R@20": 0.13854166666666665,
220
+ "R@50": 0.16895833333333332,
221
+ "R@100": 0.19395833333333332
222
+ },
223
+ "walking in": {
224
+ "R@20": 0.007042253521126761,
225
+ "R@50": 0.025821596244131453,
226
+ "R@100": 0.07276995305164319
227
+ },
228
+ "walking on": {
229
+ "R@20": 0.03820375335120643,
230
+ "R@50": 0.09658176943699734,
231
+ "R@100": 0.15212243074173368
232
+ },
233
+ "watching": {
234
+ "R@20": 0.026262626262626265,
235
+ "R@50": 0.04971139971139971,
236
+ "R@100": 0.09600769600769601
237
+ },
238
+ "wearing": {
239
+ "R@20": 0.43773830288438165,
240
+ "R@50": 0.5042531345602067,
241
+ "R@100": 0.5496789246415315
242
+ },
243
+ "wears": {
244
+ "R@20": 0.0,
245
+ "R@50": 0.0,
246
+ "R@100": 0.00013842746400885935
247
+ },
248
+ "with": {
249
+ "R@20": 0.006124588200059898,
250
+ "R@50": 0.015753219526804432,
251
+ "R@100": 0.03136168513527004
252
+ }
253
+ },
254
+ "sgdet_recall_with_constraint_mR@K": {
255
+ "mR@20": 0.0685558184377165,
256
+ "mR@50": 0.10355626064595619,
257
+ "mR@100": 0.13606674084998174,
258
+ "valid_relations": 50,
259
+ "total_relations": 50
260
+ },
261
+ "sgdet_recall_without_constraint_R@K": {},
262
+ "sgdet_recall_without_constraint_mR@K": {}
263
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.2.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "AP50": 0.3108365771999762,
3
+ "(single)R@20": 0.23070578229352812,
4
+ "(single)R@50": 0.2929217262399124,
5
+ "(single)R@100": 0.332902109339687,
6
+ "(single)mR@20": 0.08066815345015485,
7
+ "(single)mR@50": 0.12837122336340046,
8
+ "(single)mR@100": 0.16961804429088428,
9
+ "eval_arg": {
10
+ "data_path": "dataset/visual_genome",
11
+ "output_path": "./output",
12
+ "backbone_dirpath": "",
13
+ "architecture": "SenseTime/deformable-detr",
14
+ "auxiliary_loss": false,
15
+ "from_scratch": false,
16
+ "pretrained": "./ckpt/pretrained_detr__SenseTime__deformable-detr/batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune/version_0/",
17
+ "num_queries": 200,
18
+ "ce_loss_coefficient": 2.0,
19
+ "rel_loss_coefficient": 15.0,
20
+ "connectivity_loss_coefficient": 30.0,
21
+ "smoothing": 1e-14,
22
+ "rel_sample_negatives": 80,
23
+ "rel_sample_nonmatching": 80,
24
+ "rel_sample_negatives_largest": true,
25
+ "rel_sample_nonmatching_largest": true,
26
+ "batch_size": 12,
27
+ "accumulate": 2,
28
+ "gpus": 1,
29
+ "max_epochs": 70,
30
+ "max_epochs_finetune": 25,
31
+ "lr_backbone": 1e-07,
32
+ "lr": 1e-06,
33
+ "lr_initialized": 0.0001,
34
+ "weight_decay": 0.0001,
35
+ "gradient_clip_val": 0.1,
36
+ "frozen_params": false,
37
+ "debug": false,
38
+ "resume": true,
39
+ "memo": "07-27-egtr-3-NO6-fintune",
40
+ "version": 0,
41
+ "patience": 15,
42
+ "finetune": true,
43
+ "skip_main_train": true,
44
+ "main_trained_path": "output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/",
45
+ "filter_duplicate_rels": true,
46
+ "filter_multiple_rels": true,
47
+ "use_freq_bias": true,
48
+ "use_log_softmax": false,
49
+ "skip_train": true,
50
+ "split": "test",
51
+ "eval_batch_size": 3,
52
+ "eval_when_train_end": true,
53
+ "eval_single_preds": true,
54
+ "eval_multiple_preds": false,
55
+ "logit_adjustment": true,
56
+ "logit_adj_tau": 0.2,
57
+ "num_workers": 8,
58
+ "precision": 32,
59
+ "depth_data_path": "dataset/vg_depth",
60
+ "depth_file_suffix": "_depth.pt",
61
+ "use_depth_input": true,
62
+ "depth_strength_factor": 1.0,
63
+ "enable_correlation": true,
64
+ "use_depth_cross_attention": true,
65
+ "depth_fusion_heads": 8,
66
+ "depth_fusion_method": "concat",
67
+ "depth_decoder_fusion_layers": 6
68
+ },
69
+ "weight_source": "finetune",
70
+ "weight_path": "./output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695.ckpt"
71
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.2__relation_results.json ADDED
@@ -0,0 +1,263 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "sgdet_recall_with_constraint_R@K": {
3
+ "above": {
4
+ "R@20": 0.07916666666666666,
5
+ "R@50": 0.11134893713251921,
6
+ "R@100": 0.1404412999935388
7
+ },
8
+ "across": {
9
+ "R@20": 0.015873015873015872,
10
+ "R@50": 0.031746031746031744,
11
+ "R@100": 0.07936507936507936
12
+ },
13
+ "against": {
14
+ "R@20": 0.024193548387096774,
15
+ "R@50": 0.07258064516129033,
16
+ "R@100": 0.08870967741935484
17
+ },
18
+ "along": {
19
+ "R@20": 0.01834862385321101,
20
+ "R@50": 0.05045871559633028,
21
+ "R@100": 0.09021406727828747
22
+ },
23
+ "and": {
24
+ "R@20": 0.03747534516765286,
25
+ "R@50": 0.11193293885601577,
26
+ "R@100": 0.16420118343195267
27
+ },
28
+ "at": {
29
+ "R@20": 0.19589271231606997,
30
+ "R@50": 0.2843587069864442,
31
+ "R@100": 0.3301413509442706
32
+ },
33
+ "attached to": {
34
+ "R@20": 0.007370283018867925,
35
+ "R@50": 0.022680817610062895,
36
+ "R@100": 0.04036949685534591
37
+ },
38
+ "behind": {
39
+ "R@20": 0.1346401253931902,
40
+ "R@50": 0.23079508574300722,
41
+ "R@100": 0.2910219529244324
42
+ },
43
+ "belonging to": {
44
+ "R@20": 0.0023885350318471337,
45
+ "R@50": 0.006767515923566879,
46
+ "R@100": 0.02373369730057628
47
+ },
48
+ "between": {
49
+ "R@20": 0.001388888888888889,
50
+ "R@50": 0.022222222222222223,
51
+ "R@100": 0.06180555555555556
52
+ },
53
+ "carrying": {
54
+ "R@20": 0.18589147286821703,
55
+ "R@50": 0.24507751937984495,
56
+ "R@100": 0.275968992248062
57
+ },
58
+ "covered in": {
59
+ "R@20": 0.02142857142857143,
60
+ "R@50": 0.10595238095238095,
61
+ "R@100": 0.16785714285714284
62
+ },
63
+ "covering": {
64
+ "R@20": 0.04408602150537635,
65
+ "R@50": 0.07311827956989246,
66
+ "R@100": 0.14381720430107525
67
+ },
68
+ "eating": {
69
+ "R@20": 0.076158940397351,
70
+ "R@50": 0.11368653421633552,
71
+ "R@100": 0.15011037527593818
72
+ },
73
+ "flying in": {
74
+ "R@20": 0.0,
75
+ "R@50": 0.06060606060606061,
76
+ "R@100": 0.12121212121212122
77
+ },
78
+ "for": {
79
+ "R@20": 0.02459016393442623,
80
+ "R@50": 0.04498438719750195,
81
+ "R@100": 0.0961455893832943
82
+ },
83
+ "from": {
84
+ "R@20": 0.014084507042253521,
85
+ "R@50": 0.014084507042253521,
86
+ "R@100": 0.028169014084507043
87
+ },
88
+ "growing on": {
89
+ "R@20": 0.0,
90
+ "R@50": 0.008620689655172414,
91
+ "R@100": 0.017241379310344827
92
+ },
93
+ "hanging from": {
94
+ "R@20": 0.06723107569721115,
95
+ "R@50": 0.10358565737051793,
96
+ "R@100": 0.1347941567065073
97
+ },
98
+ "has": {
99
+ "R@20": 0.2965780974880082,
100
+ "R@50": 0.37805799441863924,
101
+ "R@100": 0.426011748055756
102
+ },
103
+ "holding": {
104
+ "R@20": 0.26718554224689195,
105
+ "R@50": 0.3305073522251436,
106
+ "R@100": 0.3673176550783913
107
+ },
108
+ "in": {
109
+ "R@20": 0.09776145994926483,
110
+ "R@50": 0.13684205631766608,
111
+ "R@100": 0.16560138344040784
112
+ },
113
+ "in front of": {
114
+ "R@20": 0.08156505279936242,
115
+ "R@50": 0.11915471209404264,
116
+ "R@100": 0.17024556684598524
117
+ },
118
+ "laying on": {
119
+ "R@20": 0.10247747747747747,
120
+ "R@50": 0.16666666666666666,
121
+ "R@100": 0.21509009009009009
122
+ },
123
+ "looking at": {
124
+ "R@20": 0.03480589022757698,
125
+ "R@50": 0.06091030789825971,
126
+ "R@100": 0.07409638554216867
127
+ },
128
+ "lying on": {
129
+ "R@20": 0.061224489795918366,
130
+ "R@50": 0.09693877551020408,
131
+ "R@100": 0.10204081632653061
132
+ },
133
+ "made of": {
134
+ "R@20": 0.0,
135
+ "R@50": 0.0,
136
+ "R@100": 0.0
137
+ },
138
+ "mounted on": {
139
+ "R@20": 0.0,
140
+ "R@50": 0.05416666666666667,
141
+ "R@100": 0.08437499999999999
142
+ },
143
+ "near": {
144
+ "R@20": 0.11660584500917592,
145
+ "R@50": 0.17491676532226857,
146
+ "R@100": 0.21938572999398487
147
+ },
148
+ "of": {
149
+ "R@20": 0.21916931659721156,
150
+ "R@50": 0.3145738452226006,
151
+ "R@100": 0.36492391180041533
152
+ },
153
+ "on": {
154
+ "R@20": 0.2278054939092815,
155
+ "R@50": 0.2853801628773039,
156
+ "R@100": 0.32046430691638556
157
+ },
158
+ "on back of": {
159
+ "R@20": 0.08747044917257682,
160
+ "R@50": 0.11465721040189124,
161
+ "R@100": 0.2210401891252955
162
+ },
163
+ "over": {
164
+ "R@20": 0.07593052109181142,
165
+ "R@50": 0.16079404466501243,
166
+ "R@100": 0.20748552522746072
167
+ },
168
+ "painted on": {
169
+ "R@20": 0.003448275862068966,
170
+ "R@50": 0.03793103448275863,
171
+ "R@100": 0.0839080459770115
172
+ },
173
+ "parked on": {
174
+ "R@20": 0.044101123595505616,
175
+ "R@50": 0.17455858747993577,
176
+ "R@100": 0.2585072231139647
177
+ },
178
+ "part of": {
179
+ "R@20": 0.003527336860670194,
180
+ "R@50": 0.011757789535567312,
181
+ "R@100": 0.026572604350382126
182
+ },
183
+ "playing": {
184
+ "R@20": 0.0,
185
+ "R@50": 0.045454545454545456,
186
+ "R@100": 0.045454545454545456
187
+ },
188
+ "riding": {
189
+ "R@20": 0.3054566787955492,
190
+ "R@50": 0.39935927859515896,
191
+ "R@100": 0.4674471338923166
192
+ },
193
+ "says": {
194
+ "R@20": 0.0,
195
+ "R@50": 0.0,
196
+ "R@100": 0.08333333333333333
197
+ },
198
+ "sitting on": {
199
+ "R@20": 0.11154389880952383,
200
+ "R@50": 0.16690941220238095,
201
+ "R@100": 0.20608444940476192
202
+ },
203
+ "standing on": {
204
+ "R@20": 0.03479293313069909,
205
+ "R@50": 0.08291012326916583,
206
+ "R@100": 0.12313513171225937
207
+ },
208
+ "to": {
209
+ "R@20": 0.02459016393442623,
210
+ "R@50": 0.13387978142076504,
211
+ "R@100": 0.24180327868852458
212
+ },
213
+ "under": {
214
+ "R@20": 0.15942889454339837,
215
+ "R@50": 0.23549618320610685,
216
+ "R@100": 0.2852134577325417
217
+ },
218
+ "using": {
219
+ "R@20": 0.13979166666666668,
220
+ "R@50": 0.1877083333333333,
221
+ "R@100": 0.22020833333333334
222
+ },
223
+ "walking in": {
224
+ "R@20": 0.011737089201877934,
225
+ "R@50": 0.058685446009389665,
226
+ "R@100": 0.1295774647887324
227
+ },
228
+ "walking on": {
229
+ "R@20": 0.09593386952636282,
230
+ "R@50": 0.1720509383378016,
231
+ "R@100": 0.22763628239499553
232
+ },
233
+ "watching": {
234
+ "R@20": 0.03455988455988456,
235
+ "R@50": 0.0816979316979317,
236
+ "R@100": 0.10935545935545936
237
+ },
238
+ "wearing": {
239
+ "R@20": 0.43746174195575055,
240
+ "R@50": 0.5023909042569448,
241
+ "R@100": 0.5477327961378834
242
+ },
243
+ "wears": {
244
+ "R@20": 0.0,
245
+ "R@50": 0.0,
246
+ "R@100": 0.0009689922480620156
247
+ },
248
+ "with": {
249
+ "R@20": 0.00824598183088749,
250
+ "R@50": 0.019596685634421483,
251
+ "R@100": 0.040566037735849055
252
+ }
253
+ },
254
+ "sgdet_recall_with_constraint_mR@K": {
255
+ "mR@20": 0.08066815345015485,
256
+ "mR@50": 0.12837122336340046,
257
+ "mR@100": 0.16961804429088428,
258
+ "valid_relations": 50,
259
+ "total_relations": 50
260
+ },
261
+ "sgdet_recall_without_constraint_R@K": {},
262
+ "sgdet_recall_without_constraint_mR@K": {}
263
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.3.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "AP50": 0.31083629376792,
3
+ "(single)R@20": 0.22293541252777127,
4
+ "(single)R@50": 0.27842156996614476,
5
+ "(single)R@100": 0.31196234852039123,
6
+ "(single)mR@20": 0.0973406813538091,
7
+ "(single)mR@50": 0.1533125959946274,
8
+ "(single)mR@100": 0.19882191580158376,
9
+ "eval_arg": {
10
+ "data_path": "dataset/visual_genome",
11
+ "output_path": "./output",
12
+ "backbone_dirpath": "",
13
+ "architecture": "SenseTime/deformable-detr",
14
+ "auxiliary_loss": false,
15
+ "from_scratch": false,
16
+ "pretrained": "./ckpt/pretrained_detr__SenseTime__deformable-detr/batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune/version_0/",
17
+ "num_queries": 200,
18
+ "ce_loss_coefficient": 2.0,
19
+ "rel_loss_coefficient": 15.0,
20
+ "connectivity_loss_coefficient": 30.0,
21
+ "smoothing": 1e-14,
22
+ "rel_sample_negatives": 80,
23
+ "rel_sample_nonmatching": 80,
24
+ "rel_sample_negatives_largest": true,
25
+ "rel_sample_nonmatching_largest": true,
26
+ "batch_size": 12,
27
+ "accumulate": 2,
28
+ "gpus": 1,
29
+ "max_epochs": 70,
30
+ "max_epochs_finetune": 25,
31
+ "lr_backbone": 1e-07,
32
+ "lr": 1e-06,
33
+ "lr_initialized": 0.0001,
34
+ "weight_decay": 0.0001,
35
+ "gradient_clip_val": 0.1,
36
+ "frozen_params": false,
37
+ "debug": false,
38
+ "resume": true,
39
+ "memo": "07-27-egtr-3-NO6-fintune",
40
+ "version": 0,
41
+ "patience": 15,
42
+ "finetune": true,
43
+ "skip_main_train": true,
44
+ "main_trained_path": "output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/",
45
+ "filter_duplicate_rels": true,
46
+ "filter_multiple_rels": true,
47
+ "use_freq_bias": true,
48
+ "use_log_softmax": false,
49
+ "skip_train": true,
50
+ "split": "test",
51
+ "eval_batch_size": 3,
52
+ "eval_when_train_end": true,
53
+ "eval_single_preds": true,
54
+ "eval_multiple_preds": false,
55
+ "logit_adjustment": true,
56
+ "logit_adj_tau": 0.3,
57
+ "num_workers": 8,
58
+ "precision": 32,
59
+ "depth_data_path": "dataset/vg_depth",
60
+ "depth_file_suffix": "_depth.pt",
61
+ "use_depth_input": true,
62
+ "depth_strength_factor": 1.0,
63
+ "enable_correlation": true,
64
+ "use_depth_cross_attention": true,
65
+ "depth_fusion_heads": 8,
66
+ "depth_fusion_method": "concat",
67
+ "depth_decoder_fusion_layers": 6
68
+ },
69
+ "weight_source": "finetune",
70
+ "weight_path": "./output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695.ckpt"
71
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__la_0.3__relation_results.json ADDED
@@ -0,0 +1,263 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "sgdet_recall_with_constraint_R@K": {
3
+ "above": {
4
+ "R@20": 0.08136815920398009,
5
+ "R@50": 0.12057698520385085,
6
+ "R@100": 0.1556013762357046
7
+ },
8
+ "across": {
9
+ "R@20": 0.031746031746031744,
10
+ "R@50": 0.07936507936507936,
11
+ "R@100": 0.09523809523809523
12
+ },
13
+ "against": {
14
+ "R@20": 0.056451612903225805,
15
+ "R@50": 0.12096774193548387,
16
+ "R@100": 0.1693548387096774
17
+ },
18
+ "along": {
19
+ "R@20": 0.027522935779816515,
20
+ "R@50": 0.06269113149847096,
21
+ "R@100": 0.12232415902140671
22
+ },
23
+ "and": {
24
+ "R@20": 0.0670611439842209,
25
+ "R@50": 0.15187376725838264,
26
+ "R@100": 0.22090729783037472
27
+ },
28
+ "at": {
29
+ "R@20": 0.21462750550341791,
30
+ "R@50": 0.3095817402386745,
31
+ "R@100": 0.3452265090951222
32
+ },
33
+ "attached to": {
34
+ "R@20": 0.01963443396226415,
35
+ "R@50": 0.039740566037735854,
36
+ "R@100": 0.06544811320754718
37
+ },
38
+ "behind": {
39
+ "R@20": 0.13662494734298586,
40
+ "R@50": 0.2283324931845533,
41
+ "R@100": 0.2851298584149538
42
+ },
43
+ "belonging to": {
44
+ "R@20": 0.0036965423111919925,
45
+ "R@50": 0.021003942978465272,
46
+ "R@100": 0.08475887170154686
47
+ },
48
+ "between": {
49
+ "R@20": 0.008333333333333333,
50
+ "R@50": 0.03888888888888889,
51
+ "R@100": 0.08194444444444444
52
+ },
53
+ "carrying": {
54
+ "R@20": 0.22228682170542638,
55
+ "R@50": 0.27972868217054264,
56
+ "R@100": 0.3016279069767442
57
+ },
58
+ "covered in": {
59
+ "R@20": 0.03809523809523809,
60
+ "R@50": 0.13273809523809524,
61
+ "R@100": 0.2
62
+ },
63
+ "covering": {
64
+ "R@20": 0.04408602150537635,
65
+ "R@50": 0.1021505376344086,
66
+ "R@100": 0.1696236559139785
67
+ },
68
+ "eating": {
69
+ "R@20": 0.08940397350993377,
70
+ "R@50": 0.12803532008830024,
71
+ "R@100": 0.15949227373068434
72
+ },
73
+ "flying in": {
74
+ "R@20": 0.0,
75
+ "R@50": 0.12121212121212122,
76
+ "R@100": 0.4090909090909091
77
+ },
78
+ "for": {
79
+ "R@20": 0.041803278688524584,
80
+ "R@50": 0.09908274785323966,
81
+ "R@100": 0.14430132708821233
82
+ },
83
+ "from": {
84
+ "R@20": 0.014084507042253521,
85
+ "R@50": 0.028169014084507043,
86
+ "R@100": 0.04225352112676056
87
+ },
88
+ "growing on": {
89
+ "R@20": 0.0,
90
+ "R@50": 0.034482758620689655,
91
+ "R@100": 0.04310344827586207
92
+ },
93
+ "hanging from": {
94
+ "R@20": 0.09113545816733068,
95
+ "R@50": 0.1254980079681275,
96
+ "R@100": 0.15073041168658696
97
+ },
98
+ "has": {
99
+ "R@20": 0.2928556226676025,
100
+ "R@50": 0.37014790506062645,
101
+ "R@100": 0.4183261737629751
102
+ },
103
+ "holding": {
104
+ "R@20": 0.2627785892816568,
105
+ "R@50": 0.31668662966208977,
106
+ "R@100": 0.3493319700068166
107
+ },
108
+ "in": {
109
+ "R@20": 0.09773133993377894,
110
+ "R@50": 0.1330790675449212,
111
+ "R@100": 0.15841555301067495
112
+ },
113
+ "in front of": {
114
+ "R@20": 0.09168659095437338,
115
+ "R@50": 0.14191821079896394,
116
+ "R@100": 0.19172643952978682
117
+ },
118
+ "laying on": {
119
+ "R@20": 0.1719219219219219,
120
+ "R@50": 0.22297297297297297,
121
+ "R@100": 0.26238738738738737
122
+ },
123
+ "looking at": {
124
+ "R@20": 0.038821954484605084,
125
+ "R@50": 0.06291834002677375,
126
+ "R@100": 0.0821285140562249
127
+ },
128
+ "lying on": {
129
+ "R@20": 0.09183673469387756,
130
+ "R@50": 0.12074829931972787,
131
+ "R@100": 0.15986394557823128
132
+ },
133
+ "made of": {
134
+ "R@20": 0.0,
135
+ "R@50": 0.0,
136
+ "R@100": 0.0
137
+ },
138
+ "mounted on": {
139
+ "R@20": 0.026041666666666668,
140
+ "R@50": 0.11979166666666667,
141
+ "R@100": 0.16145833333333334
142
+ },
143
+ "near": {
144
+ "R@20": 0.10750213594167252,
145
+ "R@50": 0.157036347405645,
146
+ "R@100": 0.19381506001708754
147
+ },
148
+ "of": {
149
+ "R@20": 0.2190911359012952,
150
+ "R@50": 0.30706225768581746,
151
+ "R@100": 0.3464654449575064
152
+ },
153
+ "on": {
154
+ "R@20": 0.20201358481239315,
155
+ "R@50": 0.24632449277709742,
156
+ "R@100": 0.2688159536239559
157
+ },
158
+ "on back of": {
159
+ "R@20": 0.13356973995271867,
160
+ "R@50": 0.1950354609929078,
161
+ "R@100": 0.2647754137115839
162
+ },
163
+ "over": {
164
+ "R@20": 0.08213399503722085,
165
+ "R@50": 0.17568238213399504,
166
+ "R@100": 0.21368899917287015
167
+ },
168
+ "painted on": {
169
+ "R@20": 0.019252873563218392,
170
+ "R@50": 0.07097701149425288,
171
+ "R@100": 0.11551724137931035
172
+ },
173
+ "parked on": {
174
+ "R@20": 0.1734617442482611,
175
+ "R@50": 0.2762974852862493,
176
+ "R@100": 0.36816479400749064
177
+ },
178
+ "part of": {
179
+ "R@20": 0.003527336860670194,
180
+ "R@50": 0.024103468547912992,
181
+ "R@100": 0.08212815990593768
182
+ },
183
+ "playing": {
184
+ "R@20": 0.045454545454545456,
185
+ "R@50": 0.045454545454545456,
186
+ "R@100": 0.045454545454545456
187
+ },
188
+ "riding": {
189
+ "R@20": 0.35685941043083896,
190
+ "R@50": 0.45629910879080315,
191
+ "R@100": 0.5128513420872224
192
+ },
193
+ "says": {
194
+ "R@20": 0.0,
195
+ "R@50": 0.08333333333333333,
196
+ "R@100": 0.08333333333333333
197
+ },
198
+ "sitting on": {
199
+ "R@20": 0.12050967261904762,
200
+ "R@50": 0.17978143601190474,
201
+ "R@100": 0.22182105654761908
202
+ },
203
+ "standing on": {
204
+ "R@20": 0.06648619554204661,
205
+ "R@50": 0.11270052347180007,
206
+ "R@100": 0.145931484295846
207
+ },
208
+ "to": {
209
+ "R@20": 0.07581967213114754,
210
+ "R@50": 0.21789617486338797,
211
+ "R@100": 0.305327868852459
212
+ },
213
+ "under": {
214
+ "R@20": 0.16595985298275373,
215
+ "R@50": 0.24160305343511448,
216
+ "R@100": 0.29072660446706244
217
+ },
218
+ "using": {
219
+ "R@20": 0.15854166666666666,
220
+ "R@50": 0.20645833333333333,
221
+ "R@100": 0.27020833333333333
222
+ },
223
+ "walking in": {
224
+ "R@20": 0.0516431924882629,
225
+ "R@50": 0.13380281690140844,
226
+ "R@100": 0.20469483568075117
227
+ },
228
+ "walking on": {
229
+ "R@20": 0.1296023235031278,
230
+ "R@50": 0.20580875781948169,
231
+ "R@100": 0.2384271671134942
232
+ },
233
+ "watching": {
234
+ "R@20": 0.047691197691197694,
235
+ "R@50": 0.0923039923039923,
236
+ "R@100": 0.13090428090428088
237
+ },
238
+ "wearing": {
239
+ "R@20": 0.4352361965651805,
240
+ "R@50": 0.4989630916693831,
241
+ "R@100": 0.5410597147008801
242
+ },
243
+ "wears": {
244
+ "R@20": 0.0,
245
+ "R@50": 0.0005767811000369139,
246
+ "R@100": 0.018519287559985234
247
+ },
248
+ "with": {
249
+ "R@20": 0.011041229909154437,
250
+ "R@50": 0.025746231406608765,
251
+ "R@100": 0.04866953451859112
252
+ }
253
+ },
254
+ "sgdet_recall_with_constraint_mR@K": {
255
+ "mR@20": 0.0973406813538091,
256
+ "mR@50": 0.1533125959946274,
257
+ "mR@100": 0.19882191580158376,
258
+ "valid_relations": 50,
259
+ "total_relations": 50
260
+ },
261
+ "sgdet_recall_without_constraint_R@K": {},
262
+ "sgdet_recall_without_constraint_mR@K": {}
263
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__no__la.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "AP50": 0.3108389356569485,
3
+ "(single)R@20": 0.23635742876518284,
4
+ "(single)R@50": 0.3028534447090023,
5
+ "(single)R@100": 0.3473717402840602,
6
+ "(single)mR@20": 0.05985749452580964,
7
+ "(single)mR@50": 0.08607311657799739,
8
+ "(single)mR@100": 0.10750285419405314,
9
+ "eval_arg": {
10
+ "data_path": "dataset/visual_genome",
11
+ "output_path": "./output",
12
+ "backbone_dirpath": "",
13
+ "architecture": "SenseTime/deformable-detr",
14
+ "auxiliary_loss": false,
15
+ "from_scratch": false,
16
+ "pretrained": "./ckpt/pretrained_detr__SenseTime__deformable-detr/batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune/version_0/",
17
+ "num_queries": 200,
18
+ "ce_loss_coefficient": 2.0,
19
+ "rel_loss_coefficient": 15.0,
20
+ "connectivity_loss_coefficient": 30.0,
21
+ "smoothing": 1e-14,
22
+ "rel_sample_negatives": 80,
23
+ "rel_sample_nonmatching": 80,
24
+ "rel_sample_negatives_largest": true,
25
+ "rel_sample_nonmatching_largest": true,
26
+ "batch_size": 12,
27
+ "accumulate": 2,
28
+ "gpus": 1,
29
+ "max_epochs": 70,
30
+ "max_epochs_finetune": 25,
31
+ "lr_backbone": 1e-07,
32
+ "lr": 1e-06,
33
+ "lr_initialized": 0.0001,
34
+ "weight_decay": 0.0001,
35
+ "gradient_clip_val": 0.1,
36
+ "frozen_params": false,
37
+ "debug": false,
38
+ "resume": true,
39
+ "memo": "07-27-egtr-3-NO6-fintune",
40
+ "version": 0,
41
+ "patience": 15,
42
+ "finetune": true,
43
+ "skip_main_train": true,
44
+ "main_trained_path": "output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/",
45
+ "filter_duplicate_rels": true,
46
+ "filter_multiple_rels": true,
47
+ "use_freq_bias": true,
48
+ "use_log_softmax": false,
49
+ "skip_train": true,
50
+ "split": "test",
51
+ "eval_batch_size": 3,
52
+ "eval_when_train_end": true,
53
+ "eval_single_preds": true,
54
+ "eval_multiple_preds": false,
55
+ "logit_adjustment": false,
56
+ "logit_adj_tau": 0.5,
57
+ "num_workers": 8,
58
+ "precision": 32,
59
+ "depth_data_path": "dataset/vg_depth",
60
+ "depth_file_suffix": "_depth.pt",
61
+ "use_depth_input": true,
62
+ "depth_strength_factor": 1.0,
63
+ "enable_correlation": true,
64
+ "use_depth_cross_attention": true,
65
+ "depth_fusion_heads": 8,
66
+ "depth_fusion_method": "concat",
67
+ "depth_decoder_fusion_layers": 6
68
+ },
69
+ "weight_source": "finetune",
70
+ "weight_path": "./output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695.ckpt"
71
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/epoch=06-validation_loss=1.695__test__8816__GeForce__no__la__relation_results.json ADDED
@@ -0,0 +1,263 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "sgdet_recall_with_constraint_R@K": {
3
+ "above": {
4
+ "R@20": 0.06890547263681591,
5
+ "R@50": 0.09699715707178394,
6
+ "R@100": 0.11572316986496091
7
+ },
8
+ "across": {
9
+ "R@20": 0.007936507936507936,
10
+ "R@50": 0.023809523809523808,
11
+ "R@100": 0.023809523809523808
12
+ },
13
+ "against": {
14
+ "R@20": 0.024193548387096774,
15
+ "R@50": 0.024193548387096774,
16
+ "R@100": 0.024193548387096774
17
+ },
18
+ "along": {
19
+ "R@20": 0.009174311926605505,
20
+ "R@50": 0.009174311926605505,
21
+ "R@100": 0.009174311926605505
22
+ },
23
+ "and": {
24
+ "R@20": 0.014792899408284023,
25
+ "R@50": 0.05128205128205128,
26
+ "R@100": 0.07495069033530571
27
+ },
28
+ "at": {
29
+ "R@20": 0.13007762715791912,
30
+ "R@50": 0.19319893407484648,
31
+ "R@100": 0.24287452207160232
32
+ },
33
+ "attached to": {
34
+ "R@20": 0.0,
35
+ "R@50": 0.0026533018867924527,
36
+ "R@100": 0.004127358490566038
37
+ },
38
+ "behind": {
39
+ "R@20": 0.12649846359128322,
40
+ "R@50": 0.22391621313826127,
41
+ "R@100": 0.2890438904301933
42
+ },
43
+ "belonging to": {
44
+ "R@20": 0.0,
45
+ "R@50": 0.0,
46
+ "R@100": 0.0009099181073703366
47
+ },
48
+ "between": {
49
+ "R@20": 0.0,
50
+ "R@50": 0.001388888888888889,
51
+ "R@100": 0.008333333333333333
52
+ },
53
+ "carrying": {
54
+ "R@20": 0.10523255813953489,
55
+ "R@50": 0.1522093023255814,
56
+ "R@100": 0.18310077519379844
57
+ },
58
+ "covered in": {
59
+ "R@20": 0.0,
60
+ "R@50": 0.007142857142857143,
61
+ "R@100": 0.04345238095238095
62
+ },
63
+ "covering": {
64
+ "R@20": 0.0,
65
+ "R@50": 0.012903225806451613,
66
+ "R@100": 0.012903225806451613
67
+ },
68
+ "eating": {
69
+ "R@20": 0.06622516556291391,
70
+ "R@50": 0.0739514348785872,
71
+ "R@100": 0.09050772626931566
72
+ },
73
+ "flying in": {
74
+ "R@20": 0.0,
75
+ "R@50": 0.0,
76
+ "R@100": 0.0
77
+ },
78
+ "for": {
79
+ "R@20": 0.0,
80
+ "R@50": 0.006147540983606557,
81
+ "R@100": 0.010245901639344262
82
+ },
83
+ "from": {
84
+ "R@20": 0.0,
85
+ "R@50": 0.0,
86
+ "R@100": 0.0
87
+ },
88
+ "growing on": {
89
+ "R@20": 0.0,
90
+ "R@50": 0.0,
91
+ "R@100": 0.0
92
+ },
93
+ "hanging from": {
94
+ "R@20": 0.035856573705179286,
95
+ "R@50": 0.06872509960159362,
96
+ "R@100": 0.07768924302788845
97
+ },
98
+ "has": {
99
+ "R@20": 0.3012178701174347,
100
+ "R@50": 0.38372819860828683,
101
+ "R@100": 0.4357496782985025
102
+ },
103
+ "holding": {
104
+ "R@20": 0.26346366085629885,
105
+ "R@50": 0.3322115103710196,
106
+ "R@100": 0.37318677573278797
107
+ },
108
+ "in": {
109
+ "R@20": 0.09489610958147544,
110
+ "R@50": 0.13397377554694628,
111
+ "R@100": 0.16556808178271593
112
+ },
113
+ "in front of": {
114
+ "R@20": 0.05524008766686591,
115
+ "R@50": 0.08735305837816297,
116
+ "R@100": 0.11847230524008767
117
+ },
118
+ "laying on": {
119
+ "R@20": 0.0045045045045045045,
120
+ "R@50": 0.02177177177177177,
121
+ "R@100": 0.052177177177177174
122
+ },
123
+ "looking at": {
124
+ "R@20": 0.018741633199464522,
125
+ "R@50": 0.04685408299866131,
126
+ "R@100": 0.06894243641231594
127
+ },
128
+ "lying on": {
129
+ "R@20": 0.0,
130
+ "R@50": 0.01020408163265306,
131
+ "R@100": 0.04591836734693878
132
+ },
133
+ "made of": {
134
+ "R@20": 0.0,
135
+ "R@50": 0.0,
136
+ "R@100": 0.0
137
+ },
138
+ "mounted on": {
139
+ "R@20": 0.0,
140
+ "R@50": 0.0,
141
+ "R@100": 0.0
142
+ },
143
+ "near": {
144
+ "R@20": 0.12785440348800234,
145
+ "R@50": 0.1994206138485646,
146
+ "R@100": 0.2537019796252236
147
+ },
148
+ "of": {
149
+ "R@20": 0.2076770493580653,
150
+ "R@50": 0.30236269139500255,
151
+ "R@100": 0.35507849303334355
152
+ },
153
+ "on": {
154
+ "R@20": 0.2582196946906444,
155
+ "R@50": 0.3325839237163287,
156
+ "R@100": 0.3837283568065826
157
+ },
158
+ "on back of": {
159
+ "R@20": 0.03546099290780142,
160
+ "R@50": 0.059101654846335686,
161
+ "R@100": 0.08037825059101654
162
+ },
163
+ "over": {
164
+ "R@20": 0.02977667493796526,
165
+ "R@50": 0.09636062861869313,
166
+ "R@100": 0.1346153846153846
167
+ },
168
+ "painted on": {
169
+ "R@20": 0.0,
170
+ "R@50": 0.0,
171
+ "R@100": 0.020689655172413793
172
+ },
173
+ "parked on": {
174
+ "R@20": 0.0,
175
+ "R@50": 0.0,
176
+ "R@100": 0.0
177
+ },
178
+ "part of": {
179
+ "R@20": 0.0,
180
+ "R@50": 0.0,
181
+ "R@100": 0.0
182
+ },
183
+ "playing": {
184
+ "R@20": 0.0,
185
+ "R@50": 0.045454545454545456,
186
+ "R@100": 0.045454545454545456
187
+ },
188
+ "riding": {
189
+ "R@20": 0.18306702525971627,
190
+ "R@50": 0.22175552391499237,
191
+ "R@100": 0.257359067658071
192
+ },
193
+ "says": {
194
+ "R@20": 0.0,
195
+ "R@50": 0.0,
196
+ "R@100": 0.0
197
+ },
198
+ "sitting on": {
199
+ "R@20": 0.07827566964285715,
200
+ "R@50": 0.11371186755952381,
201
+ "R@100": 0.15263764880952382
202
+ },
203
+ "standing on": {
204
+ "R@20": 0.008833586626139816,
205
+ "R@50": 0.0165273556231003,
206
+ "R@100": 0.02918671901384667
207
+ },
208
+ "to": {
209
+ "R@20": 0.0,
210
+ "R@50": 0.00819672131147541,
211
+ "R@100": 0.055327868852459015
212
+ },
213
+ "under": {
214
+ "R@20": 0.14068419564602772,
215
+ "R@50": 0.2098529827537461,
216
+ "R@100": 0.2605315238903025
217
+ },
218
+ "using": {
219
+ "R@20": 0.12604166666666666,
220
+ "R@50": 0.16145833333333331,
221
+ "R@100": 0.17395833333333333
222
+ },
223
+ "walking in": {
224
+ "R@20": 0.007042253521126761,
225
+ "R@50": 0.007042253521126761,
226
+ "R@100": 0.011737089201877934
227
+ },
228
+ "walking on": {
229
+ "R@20": 0.002680965147453083,
230
+ "R@50": 0.02680965147453083,
231
+ "R@100": 0.04349865951742628
232
+ },
233
+ "watching": {
234
+ "R@20": 0.01616161616161616,
235
+ "R@50": 0.020530703864037197,
236
+ "R@100": 0.0700537117203784
237
+ },
238
+ "wearing": {
239
+ "R@20": 0.4381670950909194,
240
+ "R@50": 0.5049898085401434,
241
+ "R@100": 0.5507342893161369
242
+ },
243
+ "wears": {
244
+ "R@20": 0.0,
245
+ "R@50": 0.0,
246
+ "R@100": 0.0
247
+ },
248
+ "with": {
249
+ "R@20": 0.005974842767295597,
250
+ "R@50": 0.01370669861235899,
251
+ "R@100": 0.025416791454527302
252
+ }
253
+ },
254
+ "sgdet_recall_with_constraint_mR@K": {
255
+ "mR@20": 0.05985749452580964,
256
+ "mR@50": 0.08607311657799739,
257
+ "mR@100": 0.10750285419405314,
258
+ "valid_relations": 50,
259
+ "total_relations": 50
260
+ },
261
+ "sgdet_recall_without_constraint_R@K": {},
262
+ "sgdet_recall_without_constraint_mR@K": {}
263
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/checkpoints/last.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1241de60e7d6c315ce88c7dc89b82df22d6328f86193a66a9a2ff644ab7ca043
3
+ size 593083428
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/config.json ADDED
@@ -0,0 +1,376 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_dropout": 0.0,
3
+ "activation_function": "relu",
4
+ "architecture": "SenseTime/deformable-detr",
5
+ "architectures": [
6
+ "DeformableDetrForObjectDetection"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "auxiliary_loss": false,
10
+ "backbone": "resnet50",
11
+ "bbox_cost": 5,
12
+ "bbox_loss_coefficient": 5,
13
+ "ce_loss_coefficient": 2.0,
14
+ "class_cost": 1,
15
+ "connectivity_loss_coefficient": 30.0,
16
+ "d_model": 256,
17
+ "decoder_attention_heads": 8,
18
+ "decoder_ffn_dim": 1024,
19
+ "decoder_layerdrop": 0.0,
20
+ "decoder_layers": 6,
21
+ "decoder_n_points": 4,
22
+ "depth_decoder_fusion_layers": 6,
23
+ "depth_fusion_heads": 8,
24
+ "depth_fusion_method": "concat",
25
+ "depth_strength_factor": 1.0,
26
+ "dice_loss_coefficient": 1,
27
+ "dilation": false,
28
+ "dropout": 0.1,
29
+ "enable_correlation": true,
30
+ "encoder_attention_heads": 8,
31
+ "encoder_ffn_dim": 1024,
32
+ "encoder_layerdrop": 0.0,
33
+ "encoder_layers": 6,
34
+ "encoder_n_points": 4,
35
+ "eos_coefficient": 0.1,
36
+ "focal_alpha": 0.25,
37
+ "freq_bias_eps": 1e-12,
38
+ "from_scratch": false,
39
+ "frozen_params": false,
40
+ "giou_cost": 2,
41
+ "giou_loss_coefficient": 2,
42
+ "id2label": {
43
+ "0": "LABEL_0",
44
+ "1": "LABEL_1",
45
+ "2": "LABEL_2",
46
+ "3": "LABEL_3",
47
+ "4": "LABEL_4",
48
+ "5": "LABEL_5",
49
+ "6": "LABEL_6",
50
+ "7": "LABEL_7",
51
+ "8": "LABEL_8",
52
+ "9": "LABEL_9",
53
+ "10": "LABEL_10",
54
+ "11": "LABEL_11",
55
+ "12": "LABEL_12",
56
+ "13": "LABEL_13",
57
+ "14": "LABEL_14",
58
+ "15": "LABEL_15",
59
+ "16": "LABEL_16",
60
+ "17": "LABEL_17",
61
+ "18": "LABEL_18",
62
+ "19": "LABEL_19",
63
+ "20": "LABEL_20",
64
+ "21": "LABEL_21",
65
+ "22": "LABEL_22",
66
+ "23": "LABEL_23",
67
+ "24": "LABEL_24",
68
+ "25": "LABEL_25",
69
+ "26": "LABEL_26",
70
+ "27": "LABEL_27",
71
+ "28": "LABEL_28",
72
+ "29": "LABEL_29",
73
+ "30": "LABEL_30",
74
+ "31": "LABEL_31",
75
+ "32": "LABEL_32",
76
+ "33": "LABEL_33",
77
+ "34": "LABEL_34",
78
+ "35": "LABEL_35",
79
+ "36": "LABEL_36",
80
+ "37": "LABEL_37",
81
+ "38": "LABEL_38",
82
+ "39": "LABEL_39",
83
+ "40": "LABEL_40",
84
+ "41": "LABEL_41",
85
+ "42": "LABEL_42",
86
+ "43": "LABEL_43",
87
+ "44": "LABEL_44",
88
+ "45": "LABEL_45",
89
+ "46": "LABEL_46",
90
+ "47": "LABEL_47",
91
+ "48": "LABEL_48",
92
+ "49": "LABEL_49",
93
+ "50": "LABEL_50",
94
+ "51": "LABEL_51",
95
+ "52": "LABEL_52",
96
+ "53": "LABEL_53",
97
+ "54": "LABEL_54",
98
+ "55": "LABEL_55",
99
+ "56": "LABEL_56",
100
+ "57": "LABEL_57",
101
+ "58": "LABEL_58",
102
+ "59": "LABEL_59",
103
+ "60": "LABEL_60",
104
+ "61": "LABEL_61",
105
+ "62": "LABEL_62",
106
+ "63": "LABEL_63",
107
+ "64": "LABEL_64",
108
+ "65": "LABEL_65",
109
+ "66": "LABEL_66",
110
+ "67": "LABEL_67",
111
+ "68": "LABEL_68",
112
+ "69": "LABEL_69",
113
+ "70": "LABEL_70",
114
+ "71": "LABEL_71",
115
+ "72": "LABEL_72",
116
+ "73": "LABEL_73",
117
+ "74": "LABEL_74",
118
+ "75": "LABEL_75",
119
+ "76": "LABEL_76",
120
+ "77": "LABEL_77",
121
+ "78": "LABEL_78",
122
+ "79": "LABEL_79",
123
+ "80": "LABEL_80",
124
+ "81": "LABEL_81",
125
+ "82": "LABEL_82",
126
+ "83": "LABEL_83",
127
+ "84": "LABEL_84",
128
+ "85": "LABEL_85",
129
+ "86": "LABEL_86",
130
+ "87": "LABEL_87",
131
+ "88": "LABEL_88",
132
+ "89": "LABEL_89",
133
+ "90": "LABEL_90",
134
+ "91": "LABEL_91",
135
+ "92": "LABEL_92",
136
+ "93": "LABEL_93",
137
+ "94": "LABEL_94",
138
+ "95": "LABEL_95",
139
+ "96": "LABEL_96",
140
+ "97": "LABEL_97",
141
+ "98": "LABEL_98",
142
+ "99": "LABEL_99",
143
+ "100": "LABEL_100",
144
+ "101": "LABEL_101",
145
+ "102": "LABEL_102",
146
+ "103": "LABEL_103",
147
+ "104": "LABEL_104",
148
+ "105": "LABEL_105",
149
+ "106": "LABEL_106",
150
+ "107": "LABEL_107",
151
+ "108": "LABEL_108",
152
+ "109": "LABEL_109",
153
+ "110": "LABEL_110",
154
+ "111": "LABEL_111",
155
+ "112": "LABEL_112",
156
+ "113": "LABEL_113",
157
+ "114": "LABEL_114",
158
+ "115": "LABEL_115",
159
+ "116": "LABEL_116",
160
+ "117": "LABEL_117",
161
+ "118": "LABEL_118",
162
+ "119": "LABEL_119",
163
+ "120": "LABEL_120",
164
+ "121": "LABEL_121",
165
+ "122": "LABEL_122",
166
+ "123": "LABEL_123",
167
+ "124": "LABEL_124",
168
+ "125": "LABEL_125",
169
+ "126": "LABEL_126",
170
+ "127": "LABEL_127",
171
+ "128": "LABEL_128",
172
+ "129": "LABEL_129",
173
+ "130": "LABEL_130",
174
+ "131": "LABEL_131",
175
+ "132": "LABEL_132",
176
+ "133": "LABEL_133",
177
+ "134": "LABEL_134",
178
+ "135": "LABEL_135",
179
+ "136": "LABEL_136",
180
+ "137": "LABEL_137",
181
+ "138": "LABEL_138",
182
+ "139": "LABEL_139",
183
+ "140": "LABEL_140",
184
+ "141": "LABEL_141",
185
+ "142": "LABEL_142",
186
+ "143": "LABEL_143",
187
+ "144": "LABEL_144",
188
+ "145": "LABEL_145",
189
+ "146": "LABEL_146",
190
+ "147": "LABEL_147",
191
+ "148": "LABEL_148",
192
+ "149": "LABEL_149"
193
+ },
194
+ "init_std": 0.02,
195
+ "init_xavier_std": 1.0,
196
+ "is_encoder_decoder": true,
197
+ "label2id": {
198
+ "LABEL_0": 0,
199
+ "LABEL_1": 1,
200
+ "LABEL_10": 10,
201
+ "LABEL_100": 100,
202
+ "LABEL_101": 101,
203
+ "LABEL_102": 102,
204
+ "LABEL_103": 103,
205
+ "LABEL_104": 104,
206
+ "LABEL_105": 105,
207
+ "LABEL_106": 106,
208
+ "LABEL_107": 107,
209
+ "LABEL_108": 108,
210
+ "LABEL_109": 109,
211
+ "LABEL_11": 11,
212
+ "LABEL_110": 110,
213
+ "LABEL_111": 111,
214
+ "LABEL_112": 112,
215
+ "LABEL_113": 113,
216
+ "LABEL_114": 114,
217
+ "LABEL_115": 115,
218
+ "LABEL_116": 116,
219
+ "LABEL_117": 117,
220
+ "LABEL_118": 118,
221
+ "LABEL_119": 119,
222
+ "LABEL_12": 12,
223
+ "LABEL_120": 120,
224
+ "LABEL_121": 121,
225
+ "LABEL_122": 122,
226
+ "LABEL_123": 123,
227
+ "LABEL_124": 124,
228
+ "LABEL_125": 125,
229
+ "LABEL_126": 126,
230
+ "LABEL_127": 127,
231
+ "LABEL_128": 128,
232
+ "LABEL_129": 129,
233
+ "LABEL_13": 13,
234
+ "LABEL_130": 130,
235
+ "LABEL_131": 131,
236
+ "LABEL_132": 132,
237
+ "LABEL_133": 133,
238
+ "LABEL_134": 134,
239
+ "LABEL_135": 135,
240
+ "LABEL_136": 136,
241
+ "LABEL_137": 137,
242
+ "LABEL_138": 138,
243
+ "LABEL_139": 139,
244
+ "LABEL_14": 14,
245
+ "LABEL_140": 140,
246
+ "LABEL_141": 141,
247
+ "LABEL_142": 142,
248
+ "LABEL_143": 143,
249
+ "LABEL_144": 144,
250
+ "LABEL_145": 145,
251
+ "LABEL_146": 146,
252
+ "LABEL_147": 147,
253
+ "LABEL_148": 148,
254
+ "LABEL_149": 149,
255
+ "LABEL_15": 15,
256
+ "LABEL_16": 16,
257
+ "LABEL_17": 17,
258
+ "LABEL_18": 18,
259
+ "LABEL_19": 19,
260
+ "LABEL_2": 2,
261
+ "LABEL_20": 20,
262
+ "LABEL_21": 21,
263
+ "LABEL_22": 22,
264
+ "LABEL_23": 23,
265
+ "LABEL_24": 24,
266
+ "LABEL_25": 25,
267
+ "LABEL_26": 26,
268
+ "LABEL_27": 27,
269
+ "LABEL_28": 28,
270
+ "LABEL_29": 29,
271
+ "LABEL_3": 3,
272
+ "LABEL_30": 30,
273
+ "LABEL_31": 31,
274
+ "LABEL_32": 32,
275
+ "LABEL_33": 33,
276
+ "LABEL_34": 34,
277
+ "LABEL_35": 35,
278
+ "LABEL_36": 36,
279
+ "LABEL_37": 37,
280
+ "LABEL_38": 38,
281
+ "LABEL_39": 39,
282
+ "LABEL_4": 4,
283
+ "LABEL_40": 40,
284
+ "LABEL_41": 41,
285
+ "LABEL_42": 42,
286
+ "LABEL_43": 43,
287
+ "LABEL_44": 44,
288
+ "LABEL_45": 45,
289
+ "LABEL_46": 46,
290
+ "LABEL_47": 47,
291
+ "LABEL_48": 48,
292
+ "LABEL_49": 49,
293
+ "LABEL_5": 5,
294
+ "LABEL_50": 50,
295
+ "LABEL_51": 51,
296
+ "LABEL_52": 52,
297
+ "LABEL_53": 53,
298
+ "LABEL_54": 54,
299
+ "LABEL_55": 55,
300
+ "LABEL_56": 56,
301
+ "LABEL_57": 57,
302
+ "LABEL_58": 58,
303
+ "LABEL_59": 59,
304
+ "LABEL_6": 6,
305
+ "LABEL_60": 60,
306
+ "LABEL_61": 61,
307
+ "LABEL_62": 62,
308
+ "LABEL_63": 63,
309
+ "LABEL_64": 64,
310
+ "LABEL_65": 65,
311
+ "LABEL_66": 66,
312
+ "LABEL_67": 67,
313
+ "LABEL_68": 68,
314
+ "LABEL_69": 69,
315
+ "LABEL_7": 7,
316
+ "LABEL_70": 70,
317
+ "LABEL_71": 71,
318
+ "LABEL_72": 72,
319
+ "LABEL_73": 73,
320
+ "LABEL_74": 74,
321
+ "LABEL_75": 75,
322
+ "LABEL_76": 76,
323
+ "LABEL_77": 77,
324
+ "LABEL_78": 78,
325
+ "LABEL_79": 79,
326
+ "LABEL_8": 8,
327
+ "LABEL_80": 80,
328
+ "LABEL_81": 81,
329
+ "LABEL_82": 82,
330
+ "LABEL_83": 83,
331
+ "LABEL_84": 84,
332
+ "LABEL_85": 85,
333
+ "LABEL_86": 86,
334
+ "LABEL_87": 87,
335
+ "LABEL_88": 88,
336
+ "LABEL_89": 89,
337
+ "LABEL_9": 9,
338
+ "LABEL_90": 90,
339
+ "LABEL_91": 91,
340
+ "LABEL_92": 92,
341
+ "LABEL_93": 93,
342
+ "LABEL_94": 94,
343
+ "LABEL_95": 95,
344
+ "LABEL_96": 96,
345
+ "LABEL_97": 97,
346
+ "LABEL_98": 98,
347
+ "LABEL_99": 99
348
+ },
349
+ "logit_adj_tau": 0.5,
350
+ "logit_adjustment": false,
351
+ "main_trained_path": "output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6/version_0/checkpoints/epoch=47-validation_loss=1.695.ckpt",
352
+ "mask_loss_coefficient": 1,
353
+ "max_position_embeddings": 1024,
354
+ "model_type": "deformable_detr",
355
+ "num_feature_levels": 4,
356
+ "num_queries": 200,
357
+ "num_rel_labels": 50,
358
+ "output_attention_states": false,
359
+ "position_embedding_type": "sine",
360
+ "pretrained": "./ckpt/pretrained_detr__SenseTime__deformable-detr/batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune/version_0/",
361
+ "rel_loss_coefficient": 15.0,
362
+ "rel_sample_negatives": 80,
363
+ "rel_sample_negatives_largest": true,
364
+ "rel_sample_nonmatching": 80,
365
+ "rel_sample_nonmatching_largest": true,
366
+ "smoothing": 1e-14,
367
+ "torch_dtype": "float32",
368
+ "transformers_version": "4.18.0",
369
+ "two_stage": false,
370
+ "two_stage_num_proposals": 300,
371
+ "use_depth_cross_attention": true,
372
+ "use_depth_fusion": true,
373
+ "use_freq_bias": true,
374
+ "use_log_softmax": false,
375
+ "with_box_refine": false
376
+ }
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1753832749.hnode003.3197765.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3eac9022d84f72a97e31cb4bd3216399f4cc5ca2a19305be26420d01dc6e658
3
+ size 748310
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1753891515.hnode003.3956668.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ecfa4d2ec3ab4cbd62a35158ea98e63a249537bf29cf51d9f7c4f7d236e77dee
3
+ size 414
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1753892532.hnode003.4069729.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1ae375b7578f0d7933340e01eb151c76dc9caf0b12027cbd27b9179c124f8bae
3
+ size 338512
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1753973183.hnode003.1067497.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d094e270ab3ceb07221a38af48ba5576709d364154c4c233cbeb9ae2a90a645
3
+ size 1814
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1753973834.hnode003.1095230.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6bd119c2904e0ec4ab7782ad0ce3772e29be04376639c55f7e1f57f67fe18d9
3
+ size 33746
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1754129498.hnode003.3387473.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a8d6afe343496ada7627995c6b5c0c7db3b76d2876fbac78f9b2ce9e185908e4
3
+ size 414
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1755449537.hnode003.2175788.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:439797e03fb63025ff198326b199ded3f2929c653405e0d2390e1401f1b8f52b
3
+ size 414
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/events.out.tfevents.1755449544.hnode003.2168874.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fab723cab4b88828b417c59f04ee5f39e130a6253988ab87911e1ab737f2891b
3
+ size 414
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/hparams.yaml ADDED
@@ -0,0 +1 @@
 
 
1
+ {}
ckpt/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6-fintune__finetune/version_0/training_args.yaml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ accumulate: 2
2
+ architecture: SenseTime/deformable-detr
3
+ auxiliary_loss: false
4
+ backbone_dirpath: ''
5
+ batch_size: 2
6
+ ce_loss_coefficient: 2.0
7
+ connectivity_loss_coefficient: 30.0
8
+ data_path: dataset/visual_genome
9
+ debug: false
10
+ depth_data_path: dataset/vg_depth
11
+ depth_decoder_fusion_layers: 6
12
+ depth_file_suffix: _depth.pt
13
+ depth_fusion_heads: 8
14
+ depth_fusion_method: concat
15
+ depth_strength_factor: 1.0
16
+ enable_correlation: true
17
+ eval_batch_size: 2
18
+ eval_multiple_preds: false
19
+ eval_single_preds: true
20
+ eval_when_train_end: true
21
+ filter_duplicate_rels: true
22
+ filter_multiple_rels: true
23
+ finetune: true
24
+ from_scratch: false
25
+ frozen_params: false
26
+ gpus: 6
27
+ gradient_clip_val: 0.1
28
+ logit_adj_tau: 0.5
29
+ logit_adjustment: false
30
+ lr: 1.0e-07
31
+ lr_backbone: 1.0e-08
32
+ lr_initialized: 1.0e-05
33
+ main_trained_path: output/egtr__batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune__version_0__/batch__24__epochs__70_25__lr__1e-07_1e-06_0.0001__07-27-egtr-3-NO6/version_0/
34
+ max_epochs: 70
35
+ max_epochs_finetune: 25
36
+ memo: 07-27-egtr-3-NO6-fintune
37
+ num_queries: 200
38
+ num_workers: 8
39
+ output_path: ./output
40
+ patience: 15
41
+ precision: 32
42
+ pretrained: ./ckpt/pretrained_detr__SenseTime__deformable-detr/batch__32__epochs__150_50__lr__1e-05_0.0001__visual_genome__finetune/version_0/
43
+ rel_loss_coefficient: 15.0
44
+ rel_sample_negatives: 80
45
+ rel_sample_negatives_largest: true
46
+ rel_sample_nonmatching: 80
47
+ rel_sample_nonmatching_largest: true
48
+ resume: true
49
+ skip_main_train: true
50
+ skip_train: false
51
+ smoothing: 1.0e-14
52
+ split: test
53
+ use_depth_cross_attention: true
54
+ use_depth_input: true
55
+ use_freq_bias: true
56
+ use_log_softmax: false
57
+ version: 0
58
+ weight_decay: 0.0001