beir-cellar / beir

A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.
http://beir.ai
Apache License 2.0
1.54k stars 182 forks source link

Allow "show_progress_bar" and "convert_to_tensor" in `DenseRetrievalExactSearch` #127

Closed xhluca closed 1 year ago

xhluca commented 1 year ago

This PR would allow enabling/disabling the progress bar when defining the DRES class, as well as specifying convert_to_tensor if needed.

thakur-nandan commented 1 year ago

Hi @xhluca, thanks for the PR. I saw this late but I already implemented it in my latest revamping of the BEIR repository code.

Thanks, Nandan