File size: 854 Bytes
ef8f3ad | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | 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 |