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
767 stars 128 forks source link

请问rocketQA封装好的dual encoder trainer,有用到Cross-batch Negatives吗? #97

Closed MozerWang closed 1 year ago

MozerWang commented 1 year ago

如题,辛苦解答

ylf-Ng commented 1 year ago

trainer目前仅支持单卡训练,没有用到cross-batch negatives。多卡训练方式可参考https://github.com/PaddlePaddle/RocketQA/tree/main/research/DuReader-Retrieval-Baseline