| IDX=0,4, | |
| export PYTHONPATH=$PYTHONPATH:./ | |
| data_path=../datasets/HICO-Det | |
| model_path=./model_weights/qwen3_8b_vl_instruct | |
| output_dir=outputs | |
| 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=2 --master_port=25005 \ | |
| tools/annotate.py \ | |
| --model-path ${model_path} \ | |
| --data-path ${data_path} \ | |
| --output-dir ${output_dir} \ |