#!/bin/sh
set -x
set -e

export PYTHONPATH="./:$PYTHONPATH"

spxshift="1"
k_mean="20"
max_spread_scale="5"
temp_pool_type="uniform"
CUDA_LAUNCH_BLOCKING=1 \
CUDA_VISIBLE_DEVICES=$1 python video_chatgpt/eval/run_ad_vegas_v2p.py \
    --model-name  /your_path/llava-v1.5-7b/ \
    --projection_path /your_path/mm_projector_7b_1.5_336px.bin \
    --conv_mode pg-video-llava \
    --gt_caption /your_path/anet_val.json \
    --output_dir output/ \
    --video_dir  /your_path/anet/train_val \
    --output_name anet_ad_vegas_spxshift${spxshift}_mss${max_spread_scale}_km${k_mean}_tpt${temp_pool_type} \
    --bs 3\
    --shift_unit $spxshift \
    --sam na \
    --temp_pool_type $temp_pool_type \
    --max_spread_scale $max_spread_scale \
    --k_means $k_mean \
    --tav \                
