ghostv1 / apex /tests /distributed /DDP /run_race_test.sh
Jagrut Thakare
v1
6d92c79
raw
history blame contribute delete
119 Bytes
#!/bin/bash
CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 ddp_race_condition_test.py