diff --git a/README.md b/README.md index 2c6d756..c39c9fb 100644 --- a/README.md +++ b/README.md @@ -10,7 +10,7 @@ Catalog: - [x] Pre-trained and finetuned checkpoints - [x] Finetuning code for Image-Text Retrieval, Image Captioning, VQA, and NLVR2 - [x] Pre-training code -- [x] Download of bootstrapped image-text datasets +- [x] Download of bootstrapped pre-training datasets ### Inference demo (Image Captioning and VQA): @@ -67,6 +67,20 @@ NLVR2 | python -m torch.distributed.run --nproc_per_node=16 --use_env train_nlvr.py +### Pre-train: +1. Prepare training json files where each json file contains a list. Each item in the list is a dictonary with two key-value pairs: {'image': path_of_image, 'caption': text_of_image}. +2. In configs/pretrain.yaml, set 'train_file' as the paths for the json files . +3. Pre-train the model using 8 A100 GPUs: +
python -m torch.distributed.run --nproc_per_node=8 --use_env pretrain.py --config ./configs/Pretrain.yaml --output_dir output/Pretrain 
+ +### Pre-training datasets download: +We provide bootstrapped pre-training datasets as json files. Each json file contains a list. Each item in the list is a dictonary with two key-value pairs: {'url': url_of_image, 'caption': text_of_image}. + +Image source | Filtered web caption | Filtered synthetic caption | Filtered synthetic caption by ViT-L +--- | :---: | :---: | :---: +CC3M+CC12M+SBU |
Download| Download| Download +LAION115M | Download| Download| Download + ### Citation If you find this code to be useful for your research, please consider citing.