search

salesforce / LAVIS

LAVIS - A One-stop Library for Language-Vision Intelligence
BSD 3-Clause "New" or "Revised" License
9.96k stars 975 forks source link

blip2 pretain dataset #469

Open jingwang97 opened 1 year ago

jingwang97 commented 1 year ago

hi，it seems like that the dataset of pretrain stage1 and stage2 mentioned in the blip2 paper contains coco,cc3m,cc12m,sbu and laion ,but the config file only include coco and vg dataset.which is true data using in the pretraining stage?

dxli94 commented 1 year ago

We use datasets mentioned in the paper.

The config file is an example. You need to add datasets you need.