dset_name=tvsum
ctx_mode=video_tef
v_feat_types=slowfast_clip
t_feat_type=clip 
results_root=results_tvsum
exp_id=exp


######## data paths
train_path=data/tvsum/tvsum_train.jsonl
eval_path=data/tvsum/tvsum_val.jsonl
eval_split_name=val

######## setup video+text features
feat_root=features/tvsum

# # video features
v_feat_dim=2048
v_feat_dirs=()
v_feat_dirs+=(${feat_root}/video_features)

# # text features
t_feat_dir=${feat_root}/query_features/ # maybe not used
t_feat_dim=512

#### training
bsz=4
lr=1e-3
enc_layers=3
dec_layers=3
t2v_layers=2
moment_layers=1
dummy_layers=3
sent_layers=1
first_layer=13
n_layers=17
event_coef=0.1
pos_coef=0
set_cost_giou=4
export CUDA_VISIBLE_DEVICES=1

######## TVSUM domain name
for dset_domain in BK BT DS FM GA MS PK PR VT VU
do
    for seed in 2018
    do
        for num_dummies in 3
        do
            for num_prompts in 1 2
            do
                PYTHONPATH=$PYTHONPATH:. python llm_epet/train.py \
                --set_cost_giou ${set_cost_giou}\
                --dset_name ${dset_name} \
                --ctx_mode ${ctx_mode} \
                --train_path ${train_path} \
                --eval_path ${eval_path} \
                --eval_split_name ${eval_split_name} \
                --v_feat_dirs ${v_feat_dirs[@]} \
                --v_feat_dim ${v_feat_dim} \
                --t_feat_dir ${t_feat_dir} \
                --t_feat_dim ${t_feat_dim} \
                --bsz ${bsz} \
                --results_root ${results_root}/${dset_domain} \
                --exp_id ${exp_id} \
                --max_q_l -1 \
                --max_v_l 1000 \
                --n_epoch 1000 \
                --lr_drop 2000 \
                --max_es_cnt -1 \
                --seed $seed \
                --lr ${lr} \
                --dset_domain ${dset_domain} \
                --enc_layers ${enc_layers} \
                --dec_layers ${dec_layers} \
                --t2v_layers ${t2v_layers} \
                --moment_layers ${moment_layers} \
                --dummy_layers ${dummy_layers} \
                --sent_layers ${sent_layers} \
                --num_dummies ${num_dummies} \
                --num_prompts ${num_prompts} \
                --total_prompts 10 \
                --event_coef ${event_coef} \
                --pos_coef ${pos_coef} \
                --first_layer ${first_layer} \
                --n_layers ${n_layers} \
                ${@:1}
            done
        done
    done
done
