File size: 1,229 Bytes
663494c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
#!/usr/bin/env bash
T=`date +%m%d%H%M`
# -------------------------------------------------- #
# Usually you only need to customize these variables #
CFG=$1 #
CKPT=$2 #
GPUS=$3 #
# -------------------------------------------------- #
GPUS_PER_NODE=$(($GPUS<8?$GPUS:8))
MASTER_PORT=${MASTER_PORT:-28596}
WORK_DIR=$(echo ${CFG%.*} | sed -e "s/configs/work_dirs/g")/
# Intermediate files and logs will be saved to UniAD/projects/work_dirs/
if [ ! -d ${WORK_DIR}logs ]; then
mkdir -p ${WORK_DIR}logs
fi
export PYTHONPATH="$(realpath "$(dirname $0)/..")":"$(realpath "$(dirname $0)/../navsim")":$PYTHONPATH
export OMP_NUM_THREADS=1
echo 'WORK_DIR: ' ${WORK_DIR}
echo 'GPUS_PER_NODE: ' ${GPUS_PER_NODE}
echo 'PYTHONPATH: ' ${PYTHONPATH}
torchrun \
--nproc_per_node=${GPUS_PER_NODE} \
--master_port=${MASTER_PORT} \
$(dirname "$0")/test.py \
$CFG \
$CKPT \
--launcher pytorch \
--eval bbox \
--show-dir ${WORK_DIR} \
--cfg-options data.test.nav_filter_path=data_loop/navtest_split/navtest_failures_filtered.yaml
2>&1 | tee ${WORK_DIR}logs/eval.$T
|