search

cjw2021 / QAHOI

Apache License 2.0
47 stars 9 forks source link

Training on vcoco dataset #9

Closed OBVIOUSDAWN closed 2 years ago

OBVIOUSDAWN commented 2 years ago

Hello @cjw2021 ! Thanks for your great works! I tried to train this model on vcoco dataset to save some time.But when i build the vcoco dataset,i meet the same error on File "/root/autodl-tmp/QAHOI-main/models/matcher.py", line 53, in forward cost_verb_class = -(out_verb_prob.matmul(tgt_verb_labels_permute) / \ RuntimeError: mat1 dim 1 must match mat2 dim 0

Do you have the training code of vcoco in feature?My vcoco dataset training followed with "https://github.com/hitachi-rd-cv/qpic" to add vcoco.py and vcocoeval.py in dataset and fix it,If you could give me any help i would be appreciate.Thank you for your help.

cjw2021 commented 2 years ago

Thanks for your interests in our work. We are now testing the code of training on v-coco. Some settings in this version are only for HICO-DET. Could you share your modification and wait for our update.

cjw2021 commented 2 years ago

I update the code for V-COCO. Please try the latest code.