# python3 calc_metric.py \
#     --config ../configs/Retrieval_flickr_train_clip_b128_caps_k=1.yaml \
#     --train_config ../configs/train/full/clip_base.yaml \
#     --aug_modal image \
#     --orig_train_file /data/dataset/dataset_json/data_rewrite/flickr30k_train_caps=1.json \
#     --aug_train_file /data/dataset/dataset_json/data_stableDiffusion/many-to-one_cap=0_useCap0_imgGenIdx0.json

config=../configs/Exp/Retrieval_flickr_train_clip_b128_caps_k=1.yaml
train_config=../configs/train/full/clip_base.yaml
aug_modal=image
orig_train_file=/data/dataset/dataset_json/data_rewrite/flickr30k_train_caps=1.json
aug_train_file=/data/dataset/dataset_json/data_stableDiffusion/many-to-one_cap=0_useCap0_imgGenIdx0.json

aug_modal_list=( 
    text
    text
    text
    text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
    # text
)
aug_train_file_list=(
    /data/dataset/dataset_json/data_idx/flickr30k_train_idx1.json
    /data/dataset/dataset_json/data_idx/flickr30k_train_idx2.json
    /data/dataset/dataset_json/data_idx/flickr30k_train_idx3.json
    /data/dataset/dataset_json/data_idx/flickr30k_train_idx4.json
    # /data/dataset/dataset_json/data_rewrite/flickr30k_train_caps=1.json
    # /data/dataset/dataset_json/data_sorted/flickr30k_train_top1.json
    # /data/dataset/dataset_json/data_sorted/flickr30k_train_top2.json
    # /data/dataset/dataset_json/data_sorted/flickr30k_train_top3.json
    # /data/dataset/dataset_json/data_sorted/flickr30k_train_top4.json
    # /data/dataset/dataset_json/EDA0.3/flickr30k_train_1.json
    # /data/dataset/dataset_json/EDA0.3/flickr30k_train_2.json
    # /data/dataset/dataset_json/EDA0.3/flickr30k_train_3.json
    # /data/dataset/dataset_json/EDA0.3/flickr30k_train_4.json
    # /data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_coco_multi_top1_limit_caps=1.json
    # /data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_coco_multi_top2_limit_caps=1.json
    # /data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_coco_multi_top3_limit_caps=1.json
    # /data/dataset/dataset_json/data_rewrite/flickr30k_train_llama_coco_multi_top4_limit_caps=1.json
    # /data/dataset/dataset_json/OFA-large-caption/flickr30k_train_0.json
    # /data/dataset/dataset_json/OFA-large-caption/flickr30k_train_1.json
    # /data/dataset/dataset_json/OFA-large-caption/flickr30k_train_2.json
    # /data/dataset/dataset_json/OFA-large-caption/flickr30k_train_3.json
)

for i in ${!aug_modal_list[@]}; do
    aug_modal=${aug_modal_list[$i]}
    aug_train_file=${aug_train_file_list[$i]}
    python3 calc_metric.py \
        --config $config \
        --train_config $train_config \
        --aug_modal $aug_modal \
        --orig_train_file $orig_train_file \
        --aug_train_file $aug_train_file
done
