MengHsir commited on
Upload folder using huggingface_hub
Browse files- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
|