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 | |