31 lines
890 B
Bash
31 lines
890 B
Bash
|
#!/usr/bin/env bash
|
||
|
|
# ------------------------------------------------------------------------
|
||
|
|
# Deformable DETR
|
||
|
|
# Copyright (c) 2020 SenseTime. All Rights Reserved.
|
||
|
|
# Licensed under the Apache License, Version 2.0 [see LICENSE for details]
|
||
|
|
# ------------------------------------------------------------------------
|
||
|
|
|
||
|
|
set -x
|
||
|
|
|
||
|
|
GPUS=$1
|
||
|
|
RUN_COMMAND=${@:2}
|
||
|
|
if [ $GPUS -lt 8 ]; then
|
||
|
|
GPUS_PER_NODE=${GPUS_PER_NODE:-$GPUS}
|
||
|
|
else
|
||
|
|
GPUS_PER_NODE=${GPUS_PER_NODE:-8}
|
||
|
|
fi
|
||
|
|
MASTER_ADDR=${MASTER_ADDR:-"127.0.0.2"}
|
||
|
|
MASTER_PORT=${MASTER_PORT:-"29500"}
|
||
|
|
NODE_RANK=${NODE_RANK:-0}
|
||
|
|
|
||
|
|
# let "NNODES=GPUS/GPUS_PER_NODE"
|
||
|
|
NNODES=1
|
||
|
|
export CUDA_VISIBLE_DEVICES=0,1,2,3
|
||
|
|
# export CUDA_LAUNCH_BLOCKING=0
|
||
|
|
python ./tools/launch.py \
|
||
|
|
--nnodes ${NNODES} \
|
||
|
|
--node_rank ${NODE_RANK} \
|
||
|
|
--master_addr ${MASTER_ADDR} \
|
||
|
|
--master_port ${MASTER_PORT} \
|
||
|
|
--nproc_per_node ${GPUS_PER_NODE} \
|
||
|
|
${RUN_COMMAND}
|