WenzhengZhang / EntQA

Pytorch implementation of EntQA paper
MIT License
60 stars 12 forks source link

ValueError np.load(ars.cands_embeds_path) #7

Closed Todaime closed 2 years ago

Todaime commented 2 years ago

Hi,

I encounter an error while using the run_retriever.py script with the downloaded precomputed candidates embeddings.

This is what is returned by the program :

all_cands_embeds = np.load(args.cands_embeds_path)
  File "/home/m/anaconda3/envs/entqa/lib/python3.8/site-packages/numpy/lib/npyio.py", line 440, in load
    return format.read_array(fid, allow_pickle=allow_pickle,
  File "/home/m/anaconda3/envs/entqa/lib/python3.8/site-packages/numpy/lib/format.py", line 787, in read_array
    array.shape = shape
ValueError: cannot reshape array of size 2336706528 into shape (5903531,1024)

Would you be so kind as to help me understand what I am not doing properly ? :)