operands could not be broadcast together with shapes (1014,) (2,)

salesforce / ALBEF

Code for ALBEF: a new vision-language pre-training method

BSD 3-Clause "New" or "Revised" License

1.45k stars 193 forks source link

operands could not be broadcast together with shapes (1014,) (2,) #138

Open ccw122 opened 2 months ago

ccw122 commented 2 months ago

Hi,Thank you for your good codes！ When I run Retrieval.py,i have finished a train stage and start to Calculating Results,i meet the question: File "/root/ccw/Retrieval.py", line 324,in evaluate copes = np.where(inds == txt2img[index])[0] ValueError: operands could not be broadcast together with shapes (1014,)(2,) Would you tell me the reason?

ccw122 commented 2 months ago

i run Retrieval.py with batch_size_train: 16 and batch_size_test: 32 in Retrieval_flickr.yaml. if it has some influences?