search

jingtaozhan / RepCONC

WSDM'22 Best Paper: Learning Discrete Representations via Constrained Clustering for Effective and Efficient Dense Retrieval
MIT License
118 stars 13 forks source link

Multi-gpu training #5

Closed WenzhengZhang closed 2 years ago

WenzhengZhang commented 2 years ago

Hi, Thanks for sharing the code! Does your code support dp or ddp training? I tried running your training script but it seemed that it only supports single-gpu. Could you please tell me how to train your model with dp or ddp?

jingtaozhan commented 2 years ago

Thank you for your interest. I planned to update the code recently to support this.

jingtaozhan commented 2 years ago

DDP is now supported.

Best, Jingtao