| image_root: '/export/share/datasets/vision/nocaps/' | |
| ann_root: 'annotation' | |
| # set pretrained as a file path or an url | |
| pretrained: 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_base_caption_capfilt_large.pth' | |
| vit: 'base' | |
| batch_size: 32 | |
| image_size: 384 | |
| max_length: 20 | |
| min_length: 5 | |
| num_beams: 3 | |
| prompt: 'a picture of ' |