From d8c301006df54d234d7b7e4abeb3f7e4dea19db8 Mon Sep 17 00:00:00 2001 From: junchen14 Date: Tue, 24 Oct 2023 06:31:52 +0300 Subject: [PATCH] add data paths --- dataset/README_MINIGPTv2_FINETUNE.md | 41 +++++++++++++++++++++++++- train_configs/minigpt_v2_finetune.yaml | 1 - 2 files changed, 40 insertions(+), 2 deletions(-) diff --git a/dataset/README_MINIGPTv2_FINETUNE.md b/dataset/README_MINIGPTv2_FINETUNE.md index f7af60d..5bc978e 100644 --- a/dataset/README_MINIGPTv2_FINETUNE.md +++ b/dataset/README_MINIGPTv2_FINETUNE.md @@ -1,11 +1,45 @@ ## Download the COCO captions, RefCOCO, RefCOCO+. RefCOCOg, visual genome, textcaps, LLaVA, gqa, AOK-VQA, OK-VQA, OCR-VQA, filtered Flickr-30k, multi-task conversation, and Unnatural instruction datasets +Download the dataset + +Image source | Download path +--- | :---: +COCO 2014 images | images captions +Visual Genome | images part1 images part2 +TextCaps | images annotations +RefCOCO | annotations +RefCOCO+ | annotations +RefCOCOg | annotations +LLaVA | Compelex reasoning Detailed description Conversation +OKVQA | annotations +AOK-VQA | annotations +OCR-VQA | annotations +Filtered Flickr-30k | images: annotations: annotations +Multi-task conversation | annotations +Filtered unnatural instruction | annotations + + +. +├── ${MINIGPTv2_DATASET} +│ ├── coco_captions +│ ├── coco_images +| ├── annotations +| ├── coco_karpathy_train.json + + ### COCO captions -- [train2017](http://images.cocodataset.org/zips/train2017.zip) + + + +Download the COCO 2014 images +- [train2014](http://images.cocodataset.org/zips/train2014.zip) + + ### Visual genome + - [part1](https://cs.stanford.edu/people/rak248/VG_100K_2/images.zip), [part2](https://cs.stanford.edu/people/rak248/VG_100K_2/images2.zip) ### TextCaps @@ -69,10 +103,14 @@ detail_23k.json, and complex_reasoning_77k.json in conversation.yaml, detail.yam - [minigpt4/configs/datasets/llava/reason.yaml](../minigpt4/configs/datasets/llava/reason.yaml) + + + ### OKVQA - [OK-VQA Input Questions](https://okvqa.allenai.org/static/data/OpenEnded_mscoco_train2014_questions.json.zip) - [OK-VQA Annotations](https://okvqa.allenai.org/static/data/mscoco_train2014_annotations.json.zip) + - [okvqa_train](https://storage.googleapis.com/sfr-vision-language-research/LAVIS/datasets/okvqa/okvqa_train.json) - Images are from COCO @@ -89,6 +127,7 @@ curl -fsSL https://prior-datasets.s3.us-east-2.amazonaws.com/aokvqa/aokvqa_v1p0. ### filtered Flickr-30k + ### Multi-task conversation ### Unnatural instruction diff --git a/train_configs/minigpt_v2_finetune.yaml b/train_configs/minigpt_v2_finetune.yaml index 4039ea6..89d595a 100644 --- a/train_configs/minigpt_v2_finetune.yaml +++ b/train_configs/minigpt_v2_finetune.yaml @@ -11,7 +11,6 @@ model: ckpt: "/ibex/project/c2090/minigpt4_ckpt/448_perforamnce_correct_v10_vg/20230925064/checkpoint_32.pth" use_grad_checkpoint: True chat_template: True - # wanda_log: False lora_r: 64 lora_alpha: 16