Files
insightface/recognition/partial_fc/pytorch/run.sh
xiaoyang-coder 289a6f0925 Update run.sh
2020-11-13 23:13:48 +08:00

16 lines
505 B
Bash

export OMP_NUM_THREADS=4
starttime=`date +'%Y-%m-%d %H:%M:%S'`
# /usr/bin/
python3 -m torch.distributed.launch --nproc_per_node=8 --nnodes=1 --node_rank=0 --master_addr="127.0.0.1" --master_port=1234 partial_fc.py | tee hist.log
endtime=`date +'%Y-%m-%d %H:%M:%S'`
start_seconds=$(date --date="$starttime" +%s)
end_seconds=$(date --date="$endtime" +%s)
echo "运行时间:"$((end_seconds-start_seconds))"s"
ps -ef | grep "partial_fc" | grep -v grep | awk '{print "kill -9 "$2}' | sh