forked from SlongLiu/query2labels
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.json
39 lines (39 loc) · 1.2 KB
/
config.json
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
{
"img_size": 384,
"images_path": "/home/zhuhao/dataset/public/coco/val2017",
"annos_file": "/home/zhuhao/dataset/public/coco/annotations_trainval2017/annotations/instances_val2017.json",
"labels_file": "/home/zhuhao/myCode/vsass/stream_index/image_label/query2labels/data/coco/val_label_vectors_coco14.npy",
"config": "/home/zhuhao/myModel/query2labels/Q2L-CvT_w24-384/config_new.json",
"output": "/home/zhuhao/myCode/vsass/stream_index/image_label/query2labels",
"model_file": "/home/zhuhao/myModel/query2labels/Q2L-CvT_w24-384/checkpoint.pkl",
"num_class": 80,
"workers": 8,
"eps": 1e-05,
"pretrained": false,
"batch_size": 1,
"print_freq": 10,
"world_size": 1,
"rank": 0,
"local_rank": 0,
"amp": false,
"dist_url": "tcp://127.0.0.1:3451",
"position_embedding": "sine",
"keep_other_self_attn_dec": false,
"keep_first_self_attn_dec": false,
"keep_input_proj": false,
"backbone": "CvT_w24",
"gamma_pos": 0.0,
"gamma_neg": 0.0,
"asl_clip": 0.0,
"weight_decay": 0.01,
"ema_decay": 0.9997,
"optim": "AdamW",
"orid_norm": false,
"enc_layers": 1,
"dec_layers": 2,
"dim_feedforward": 4096,
"hidden_dim": 1024,
"dropout": 0.1,
"nheads": 4,
"pre_norm": false
}