#!/usr/bin/env bash

echo $PATH

# >>> conda initialize >>>
# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/cpfs01/user/litianyu/opt/miniconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
    eval "$__conda_setup"
else
    if [ -f "/cpfs01/user/litianyu/opt/miniconda3/etc/profile.d/conda.sh" ]; then
        . "/cpfs01/user/litianyu/opt/miniconda3/etc/profile.d/conda.sh"
    else
        export PATH="/cpfs01/user/litianyu/opt/miniconda3/bin:$PATH"
    fi
fi
unset __conda_setup
# <<< conda initialize <<<
conda activate paradrive

echo $PATH

CODE_PATH="/cpfs01/user/litianyu/projects/paradrive"
cd ${CODE_PATH}

AUTORESUME=0
CFG=configs/paradrive/navsim_openscenes_nuplan/base_e2e_onestage_r50_v2t1_baseline.py
WORK_DIR=${CODE_PATH}/work_dirs/uniad_12nodes_lr1x/

if [ ! -d ${WORK_DIR}logs ]; then
    mkdir -p ${WORK_DIR}logs
fi

export PYTHONPATH="${CODE_PATH}":$PYTHONPATH
export PYTHONPATH="${CODE_PATH}/external":$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:"${CODE_PATH}/external/toolbox"

export MASTER_PORT=10010

python -m torch.distributed.launch \
    --nproc_per_node=8 \
    --master_port=$MASTER_PORT \
    $(dirname "$0")/test.py \
    $CFG \
    ${WORK_DIR}latest.pth \
    --launcher pytorch ${@:4} \
    --out ./tmp/results.pkl \
    --eval bbox \
    --show-dir ${WORK_DIR} \
    2>&1 | tee ${WORK_DIR}logs/eval.$T