export HF_HOME=/efs/vkvermaa/2025/diffusion/latent_correct/hf_cache
Dataset_Name="pickapic" # "coco" or "pickapic"
Grad_Config="one_step_rectification_config"
# constant cosine_nesterov low_to_high_momentum  high_to_low_momentum
# low_to_high_nesterov high_to_low_nesterov
Model_Variant="spo" #lpo, spo
GPU_ID=$(nvidia-smi --query-gpu=index,memory.used --format=csv,noheader,nounits | sort -k2 -n | head -n1 | cut -d',' -f1)

python eval.py \
    --model_variant "$Model_Variant" \
    --dataset_type "$Dataset_Name" \
    --grad_config "$Grad_Config" \
    --metrics fid clip aesthetic pickscore hpsv2 hpsv21 imagereward \
    --max_samples 500 \
    --num_steps 20 \
    --cfg_scale 3 \
    --output_dir "RESULTS/$Dataset_Name/${Grad_Config}_${Model_Variant}" \
    --cuda $GPU_ID \
    --mode "gradient_ascent"  # gradient_ascent baseline  # /baseline