File size: 260 Bytes
a22eb82
 
1
2
3
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 --nnodes=1 --node_rank=0 --master_addr="127.0.0.1" --master_port=1234 train.py configs/ms1mv3_r50
ps -ef | grep "train" | grep -v grep | awk '{print "kill -9 "$2}' | sh