| 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 |