CUDA_VISIBLE_DEVICES=3 accelerate launch --mixed_precision bf16 src/generate.py \
   --dataset 'discriminator' \
   --blip-model-name 'blip2_cir_full' \
   --num-epochs 2 \
   --num-workers 4 \
   --learning-rate '2e-5' \
   --batch-size 128 \
   --transform targetpad \
   --target-ratio 1.25  \
   --save-training \
   --save-best \
   --validation-frequency 1