num_iters=2
# step_size=2.0
step_size=1.0
epochs=5
batch_size=128
total_steps=5000

output_dir="../train_results/0815_pretrained/"
config=./configs/ICLR/coco/caps_k=1.yaml
mark=pretrained

python3 -m torch.distributed.launch --nproc_per_node=1 --use_env --master_port=$RANDOM \
    train_clip_clean.py \
    --config $config \
    --model CLIP_ViT-B-16 \
    --model_name CLIP_ViT-B-16 \
    --train_config ./configs/train/full/clip_base.yaml \
    --output_dir $output_dir \
    --mark $mark \
    --evaluate
