Files
insightface/recognition/arcface_torch/run.sh
2021-07-23 15:53:39 +08:00

3 lines
260 B
Bash

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