Error in beam search - Githubissues

Pay20Y / SEED

164 stars 40 forks source link

Error in beam search #52

Closed JamesYang568 closed 2 years ago

JamesYang568 commented 2 years ago

I got RuntimeError: "index_select_out_cuda_impl" not implemented for 'Float'. The data type of t_predecessors and t_predecessors are Float32. Is this correct? And if I change the type to long(), I got a 0 Emebedding vector cos similarity.

YooWang commented 2 years ago

I also encountered the same problem, but I changed to long() type and got the result successfully.

JamesYang568 commented 2 years ago

I also encountered the same problem, but I changed to long() type and got the result successfully.

Thanks for your prompt reply! I want to futher confirm that can you get both similarity and accuracy correctly?