-
Notifications
You must be signed in to change notification settings - Fork 2
/
config.yaml
73 lines (73 loc) · 2.52 KB
/
config.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
dataset:
image_dir: '/tmp/datasets/vg/images'
annot_dir: '/tmp/datasets/vg_scene_graph_annot'
raw_annot_dir: '/tmp/datasets/vg/annotations/'
annotation_train: '/tmp/datasets/vg/annotations/instances_vg_train.json'
annotation_test: '/tmp/datasets/vg/annotations/instances_vg_test.json'
object_alias: 'datasets/vg_scene_graph_annot/object_alias.txt'
predicate_alias: 'datasets/vg_scene_graph_annot/predicate_alias.txt'
object_types: 'datasets/vg_scene_graph_annot/object_types.txt'
object_list: 'datasets/vg_scene_graph_annot/object_list.txt'
sub2super_cat_dict: 'datasets/vg_scene_graph_annot/sub2super_cat_dict.pt'
train_triplets: 'datasets/vg_scene_graph_annot/train_triplets.pt'
test_triplets: 'datasets/vg_scene_graph_annot/test_triplets.pt'
zero_shot_triplets: 'datasets/vg_scene_graph_annot/zero_shot_triplets.pt'
area_frac_thresh: 0.002
train_test_split: 0.7
percent_train: 1 #0.001
percent_test: 1 #0.002
dataset: 'vg' # vg, oiv6
supcat_clustering: 'motif' # motif, gpt2, bert, clip
return_annot: True
models:
detr101_key_before: 'datasets/vg_scene_graph_annot/detr101_key_before.txt'
detr101_key_after: 'datasets/vg_scene_graph_annot/detr101_key_after.txt'
detr101_pretrained_vg: 'checkpoints/detr101_vg_ckpt.pth'
detr101_pretrained_oiv6: 'checkpoints/detr101_oiv6_ckpt.pth'
detr_or_faster_rcnn: 'detr'
image_size: 1024
feature_size: 32
num_img_feature: 256
embed_hidden_dim: 256
num_classes: 150
num_relations: 50
num_super_classes: 17
hidden_dim: 128
faster_rcnn_hidden_dim: 2048
use_depth: True
depth_model_type: 'DPT_Large'
hierarchical_pred: False
topk_cat: 2
nms: 0.5
num_geometric: 15
num_possessive: 11
num_semantic: 24
num_negatives: 49
llm_model: 'gpt3.5' # gpt3.5, gpt4v
training:
run_mode: 'train' # train, eval, prepare_cs, train_cs, eval_cs
eval_mode: 'pc' # pc sgc or sgd
save_vis_results: False
learning_rate: 0.00001 #0.00001
weight_decay: 0.0001
batch_size: 12
num_epoch: 3
checkpoint_path: 'checkpoints/'
result_path: 'results/'
continue_train: False
start_epoch: 0
scheduler_param1: 2
scheduler_param2: 5
test_epoch: 2
load_model: True
lambda_sparsity: 0
lambda_contrast: 1
lambda_connectivity: 0.1
lambda_not_connected: 1
lambda_commonsense: 1
lambda_cs_weak: 0.1
lambda_cs_strong: 10
print_freq: 100
eval_freq: 100
print_freq_test: 20
eval_freq_test: 1