albef
copied
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
33 lines
660 B
33 lines
660 B
train_file: ['data/refcoco+_train.json']
|
|
test_file: ['data/refcoco+_val.json','data/refcoco+_test.json']
|
|
|
|
refcoco_data: 'data'
|
|
det_file: 'data/refcoco+/dets.json'
|
|
coco_file: 'data/refcoco+/cocos.json'
|
|
|
|
image_root: '/export/share/datasets/vision/coco/images/'
|
|
|
|
bert_config: 'configs/config_bert.json'
|
|
|
|
image_res: 384
|
|
batch_size: 32
|
|
|
|
queue_size: 65536
|
|
momentum: 0.995
|
|
vision_width: 768
|
|
embed_dim: 256
|
|
temp: 0.07
|
|
|
|
alpha: 0.4
|
|
distill: True
|
|
warm_up: True
|
|
|
|
optimizer: {opt: adamW, lr: 1e-5, weight_decay: 0.02}
|
|
schedular: {sched: cosine, lr: 1e-5, epochs: 5, min_lr: 1e-6, decay_rate: 1, warmup_lr: 1e-5, warmup_epochs: 1, cooldown_epochs: 0}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|