python3 -m torch.distributed.launch --nproc_per_node=4 --nnodes=1 --node_rank=0 embed.py