search

facebookresearch / multihop_dense_retrieval

Multi-hop dense retrieval for question answering
Other
214 stars 22 forks source link

Distributed Training #27

Open NehaNishikant opened 2 years ago

NehaNishikant commented 2 years ago

Hi, does train_mhop.py support distributed training?

I noticed a call to torch.distributed.init_process_group, but n_gpu is hardcoded to 1.