Spaces:
Runtime error
Runtime error
Upload vqa.yaml
Browse files- BLIP/configs/vqa.yaml +25 -0
BLIP/configs/vqa.yaml
ADDED
|
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
vqa_root: '/export/share/datasets/vision/VQA/Images/mscoco/' #followed by train2014/
|
| 2 |
+
vg_root: '/export/share/datasets/vision/visual-genome/' #followed by image/
|
| 3 |
+
train_files: ['vqa_train','vqa_val','vg_qa']
|
| 4 |
+
ann_root: 'annotation'
|
| 5 |
+
|
| 6 |
+
# set pretrained as a file path or an url
|
| 7 |
+
pretrained: 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_base_vqa_capfilt_large.pth'
|
| 8 |
+
|
| 9 |
+
# size of vit model; base or large
|
| 10 |
+
vit: 'base'
|
| 11 |
+
batch_size_train: 16
|
| 12 |
+
batch_size_test: 32
|
| 13 |
+
vit_grad_ckpt: False
|
| 14 |
+
vit_ckpt_layer: 0
|
| 15 |
+
init_lr: 2e-5
|
| 16 |
+
|
| 17 |
+
image_size: 480
|
| 18 |
+
|
| 19 |
+
k_test: 128
|
| 20 |
+
inference: 'rank'
|
| 21 |
+
|
| 22 |
+
# optimizer
|
| 23 |
+
weight_decay: 0.05
|
| 24 |
+
min_lr: 0
|
| 25 |
+
max_epoch: 10
|