| # ------------------------------------------------------------------------ | |
| # Copyright (c) 2022 megvii-research. All Rights Reserved. | |
| # ------------------------------------------------------------------------ | |
| # Modified from Deformable DETR (https://github.com/fundamentalvision/Deformable-DETR) | |
| # Copyright (c) 2020 SenseTime. All Rights Reserved. | |
| # ------------------------------------------------------------------------ | |
| # Modified from DETR (https://github.com/facebookresearch/detr) | |
| # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved | |
| # ------------------------------------------------------------------------ | |
| 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.1"} | |
| MASTER_PORT=${MASTER_PORT:-"29500"} | |
| NODE_RANK=${NODE_RANK:-0} | |
| let "NNODES=GPUS/GPUS_PER_NODE" | |
| python3 ./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} |