This repository has been archived by the owner on Mar 1, 2024. It is now read-only.
(BLINK Zeshel) A bug in biencoder/nn_prediction ? #95
Comments
Open
Sign up for free
to subscribe to this conversation on GitHub.
Already have an account?
Sign in.
L96 in blink/biencoder/nn_prediction.py should be changed from
cur_candidates = candidate_pool[src][inds]TO
cur_candidates = candidate_pool[srcs[i].item()][inds]The text was updated successfully, but these errors were encountered: