Fix sync issue when using pretrained embeddings.
@@ -1172,10 +1172,6 @@ class TorchAgent(ABC, Agent):
:param emb_type:
pretrained embedding type
"""
- if not is_primary_worker():
- # we're in distributed mode, copying embeddings in the workers
- # slows things down considerably
- return
embs, name = self._get_embtype(emb_type)
cnt = 0
for w, i in self.dict.tok2ind.items():
