PaddlePaddle / RocketQA

🚀 RocketQA, dense retrieval for information retrieval and question answering, including both Chinese and English state-of-the-art models.
Apache License 2.0
766 stars 128 forks source link

encode_query返回的迭代器遍历太慢 #115

Open cowarder opened 1 year ago

cowarder commented 1 year ago

这个函数返回的是一个迭代器,如果输入query只有一条的话,取迭代器里面的数据也需要采用大概2s的耗时,请问这个有办法优化吗