# num_iters=1
# step_size=2
num_iters=5
step_size=0.5
epochs=5

txt_sup_k=1
is_txt_aug=False
txt_aug=rand
scale_ver=0

# eval_ckpt_path=/data/train_results/flickr30k/CLIP_ViT-B-16/_bs64sgd_lr0.0001_ep5_wd0.0001_warmup0.3/MMA-iters1-step2.0-scale0-img-unsup0.0-sup1.0/2024-03-24_09-28-19/checkpoint_last.pth
eval_ckpt_path=/data/train_results/flickr30k/CLIP_ViT-B-16/_bs64sgd_lr0.0001_ep5_wd0.0001_warmup0.3/MMA-iters1-step2.0-scale0-img-unsup0.0-sup1.0/2024-03-24_09-28-19/checkpoint_ep0.pth

alpha=0.3
python3 -m torch.distributed.launch --nproc_per_node=1 --use_env --master_port=$RANDOM \
    compare_adv_attack.py \
    --config ./configs/Retrieval_flickr_train_clip_b64.yaml \
    --model CLIP_ViT-B-16 \
    --model_name CLIP_ViT-B-16 \
    --train_config ./configs/train/full/clip_base.yaml \
    --attack MMA \
    --step_size $step_size \
    --num_iters $num_iters \
    --scale_ver $scale_ver \
    --is_use_gt_caps False \
    --txt_sup_k $txt_sup_k \
    --alpha_sr $alpha \
    --alpha_ri $alpha \
    --alpha_rs $alpha \
    --p_rd $alpha \
    --alpha_unsup 0 \
    --alpha_sup 1 \
    --is_txt_aug $is_txt_aug \
    --txt_aug $txt_aug \
    --epochs $epochs \
    --eval_ckpt_path $eval_ckpt_path
