File size: 562 Bytes
73df34b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | IDX=1
export PYTHONPATH=$PYTHONPATH:./
data_path=../datasets/HICO-Det
model_path=./model_weights/qwen3_8b_vl_instruct
annotation_path=./outputs/merged_labels.json
output_dir=outputs/examiner
if [ -d ${output_dir} ]; then
echo "dir already exists"
else
mkdir -p ${output_dir}
fi
CUDA_VISIBLE_DEVICES=$IDX OMP_NUM_THREADS=1 torchrun --nnodes=1 --nproc_per_node=1 --master_port=25007 \
tools/examine_hico.py \
--model-path ${model_path} \
--data-path ${data_path} \
--annotation-path ${annotation_path} \
--output-dir ${output_dir} \
|