elq/biencoder/data_process.py (2 lines): - line 249: # TODO SORT EVERYTHING - line 317: # "pruned_ents": [1 for i in range(len(all_mentions)) if i < len(mention_idxs) else 0], # pruned last N entities, TODO change if changed blink/biencoder/biencoder.py (2 lines): - line 143: # TODO: why do we need cpu here? - line 198: # TODO: add parameters? blink/biencoder/eval_biencoder.py (1 line): - line 99: # TODO: add multiple thread process elq/common/params.py (1 line): - line 269: # TODO DELETE LATER!!! blink/biencoder/train_biencoder.py (1 line): - line 171: # TODO: reduce duplicated code here