In https://github.com/castorini/pyserini/blob/master/pyserini/encode/_auto.py though `AutoDocumentEncoder` allows batch encoding, but `AutoQueryEncoder` encodes one query at a time. This forces `FaissSearcher` # `batch_search` function to encode one query at a time https://github.com/castorini/pyserini/blob/9815d56eb4e41a62d59e41cbd49af25c6a907338/pyserini/search/faiss/_searcher.py#L194 ``` def _enc(q): if isinstance(q, dict): return self.query_encoder.encode(**q) assert isinstance(q, str) return self.query_encoder.encode(q) q_embs = [_enc(q) for q in queries] ``` Allowing batch encoding for queries would improve performance of `batch_search`
Activity