File size: 292 Bytes
d57daa8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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
|