microsoft/TAP

A KEYERROR need to be solved--Emergency!

MrLianSYSU opened this issue · 0 comments

error raise:
File "/home/lianjunliang/anaconda3/envs/TAP/pythia/datasets/vqa/m4c_textvqa/dataset.py", line 112, in load_item
[self.object_clsname[x] for x in features['image_info_0']['objects']]
KeyError: 'objects'

we print fearures,here is what it comes:
2022-07-20T19:53:09 INFO: Starting training...
******************** {'image_feature_0': tensor([[0.6171, 0.5207, 0.0000, ..., 6.4700, 0.0000, 0.0000],
[0.0000, 0.0000, 0.0000, ..., 2.4846, 0.0000, 0.0000],
[0.0000, 0.0000, 1.2065, ..., 0.0000, 0.0000, 0.0000],
...,
[0.0000, 0.0000, 1.0070, ..., 5.8578, 5.1698, 0.0000],
[4.3617, 0.0000, 0.0000, ..., 0.0000, 7.6311, 0.0000],
[0.0000, 0.0000, 1.4914, ..., 0.0000, 3.2282, 0.0000]]), 'image_info_0': {'max_features': tensor(100)}, 'image_feature_1': tensor([[0.6171, 0.5207, 0.0000, ..., 6.4700, 0.0000, 0.0000],
[0.0000, 0.0000, 0.0000, ..., 2.4846, 0.0000, 0.0000],
[0.0000, 0.0000, 1.2065, ..., 0.0000, 0.0000, 0.0000],
...,
[0.0000, 0.0000, 1.0070, ..., 5.8578, 5.1698, 0.0000],
[4.3617, 0.0000, 0.0000, ..., 0.0000, 7.6311, 0.0000],
[0.0000, 0.0000, 1.4914, ..., 0.0000, 3.2282, 0.0000]]), 'image_info_1': {'max_features': tensor(100)}}
2022-07-20T19:53:11 ERROR: Caught KeyError in DataLoader worker process 0.