File size: 459 Bytes
3a1265d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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} \ |