KeyError: 'image_id'

[mshmoon]

Hi, I want to train a blip2 model on coco caption dataset. So,I run program by <<python -m torch.distributed.run --nproc_per_node=1 train.py --cfg-path lavis/projects/blip/train/retrieval_coco_ft.yaml >>

But, It has error :

I do know the reason, is there somebody can help me?

Thanks !