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
689 B
33 lines
689 B
2 years ago
|
train_file: ['data/vqa_train.json',
|
||
|
|
'data/vqa_val.json',
|
||
|
|
'data/vg_qa.json']
|
||
|
|
|
||
|
|
test_file: ['data/vqa_test.json']
|
||
|
|
answer_list: 'data/answer_list.json'
|
||
|
|
|
||
|
|
vqa_root: '/export/share/datasets/vision/VQA/Images/mscoco/' #train2014/
|
||
|
|
vg_root: '/export/share/datasets/vision/visual-genome/' #image/
|
||
|
|
|
||
|
|
image_res: 384
|
||
|
|
batch_size_train: 32
|
||
|
|
batch_size_test: 16
|
||
|
|
k_test: 128
|
||
|
|
|
||
|
|
alpha: 0.4
|
||
|
|
distill: True
|
||
|
|
warm_up: True
|
||
|
|
|
||
|
|
eos: '[SEP]'
|
||
|
|
|
||
|
|
bert_config: 'configs/config_bert.json'
|
||
|
|
|
||
|
|
optimizer: {opt: adamW, lr: 2e-5, weight_decay: 0.02}
|
||
|
|
schedular: {sched: cosine, lr: 2e-5, epochs: 8, min_lr: 1e-6, decay_rate: 1, warmup_lr: 1e-5, warmup_epochs: 4, cooldown_epochs: 0}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|