AutoLLMAnnotation / scripts /examine_hico.sh
ayh015's picture
Update modifed code
73df34b
raw
history blame contribute delete
562 Bytes
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} \