vlmeval / eval.sh
tuandunghcmut's picture
.
d57daa8
MODEL_NAME=$1
PORT=$2
eval "$(conda shell.bash hook)"
conda activate vlmeval
torchrun \
--nproc_per_node=4 \
--rdzv_endpoint=localhost:$PORT \
--rdzv_id=4 --rdzv_backend=c10d --nnodes=1 \
run.py --verbose --reuse \
--data COCO_VAL \
--model $MODEL_NAME
# cp ./cuda-11.8/bin ./cuda/bin