MLVLab/Human_Object_Interaction

1#!/usr/bin/env bash2# ------------------------------------------------------------------------3# Deformable DETR4# Copyright (c) 2020 SenseTime. All Rights Reserved.5# Licensed under the Apache License, Version 2.0 [see LICENSE for details]6# ------------------------------------------------------------------------7 8set -x9 10GPUS=$111RUN_COMMAND=${@:2}12if [ $GPUS -lt 8 ]; then13    GPUS_PER_NODE=${GPUS_PER_NODE:-$GPUS}14else15    GPUS_PER_NODE=${GPUS_PER_NODE:-8}16fi17MASTER_ADDR=${MASTER_ADDR:-"127.0.0.1"}18MASTER_PORT=${MASTER_PORT:-"29500"}19NODE_RANK=${NODE_RANK:-0}20 21let "NNODES=GPUS/GPUS_PER_NODE"22 23python ./tools/launch.py \24    --nnodes ${NNODES} \25    --node_rank ${NODE_RANK} \26    --master_addr ${MASTER_ADDR} \27    --master_port ${MASTER_PORT} \28    --nproc_per_node ${GPUS_PER_NODE} \29    ${RUN_COMMAND}