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