File size: 471 Bytes
e3044d8 3a1265d e3044d8 3a1265d e3044d8 3a1265d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | IDX=1,
export PYTHONPATH=$PYTHONPATH:./
data_path=../datasets/coco
model_path=./model_weights/qwen3_30b_vl_instruct/models
output_dir=outputs/coco
if [ -d ${output_dir} ];then
echo "dir already exists"
else
mkdir ${output_dir}
fi
CUDA_VISIBLE_DEVICES=$IDX OMP_NUM_THREADS=1 torchrun --nnodes=1 --nproc_per_node=1 --master_port=25006 \
tools/annotate_coco.py \
--model-path ${model_path} \
--data-path ${data_path} \
--output-dir ${output_dir} \ |