Skip to content

Commit

Permalink
Update NLVR_pretrain.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
Junnan Li authored Jan 18, 2022
1 parent 8cf0593 commit 10701a9
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions configs/NLVR_pretrain.yaml
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
train_file: ['/export/home/project/VL/dataset/caption/coco_karpathy_train.json',
'/export/home/project/VL/dataset/caption/vg_caption.json',
'/export/home/project/VL/dataset/pretrain_caption/conceptual_caption_train.json',
'/export/home/project/VL/dataset/pretrain_caption/conceptual_caption_val.json',
'/export/home/project/VL/dataset/pretrain_caption/sbu_caption.json'
train_file: ['data/coco.json',
'data/vg.json',
'data/cc3m_train.json',
'data/cc3m_val.json',
'data/sbu.json'
]

# each train_file (json) contains a python list where each item is {'image': img_path, 'caption': text or list_of_text }
Expand Down

0 comments on commit 10701a9

Please sign in to comment.