#!/bin/bash

# Execute all evaluation tasks in parallel
# Each command runs in the background using &

echo "Starting all evaluation tasks in parallel..."

# Reference batch path
REF_BATCH="/gemini/space/zhaozy/zhy/dataset/VIRTUAL_imagenet256_labeled.npz"

# Base directory for sample files
SAMPLE_DIR="/gemini/space/zhaozy/zhy/gzy_new/Noise_Matching/Rectified-Noise/last_samples_depth_2_gvp_0.5"

# Change to the project root directory
cd /gemini/space/zhaozy/zhy/gzy_new/Noise_Matching

# Evaluate threshold 0.0 on GPU 0
CUDA_VISIBLE_DEVICES=0 nohup python evaluator.py \
    --ref_batch ${REF_BATCH} \
    --sample_batch ${SAMPLE_DIR}/depth-mu-2-threshold-0.0-0550000-base-cfg-1.0-64-SDE-100-Euler-sigma-Mean-0.04.npz \
    > eval_threshold_0.0.log 2>&1 &

# Evaluate threshold 0.15 on GPU 1
CUDA_VISIBLE_DEVICES=1 nohup python evaluator.py \
    --ref_batch ${REF_BATCH} \
    --sample_batch ${SAMPLE_DIR}/depth-mu-2-threshold-0.15-0550000-base-cfg-1.0-64-SDE-100-Euler-sigma-Mean-0.04.npz \
    > eval_threshold_0.15.log 2>&1 &

# Evaluate threshold 0.25 on GPU 2
CUDA_VISIBLE_DEVICES=2 nohup python evaluator.py \
    --ref_batch ${REF_BATCH} \
    --sample_batch ${SAMPLE_DIR}/depth-mu-2-threshold-0.25-0550000-base-cfg-1.0-64-SDE-100-Euler-sigma-Mean-0.04.npz \
    > eval_threshold_0.25.log 2>&1 &

# Evaluate threshold 0.5 on GPU 3
CUDA_VISIBLE_DEVICES=3 nohup python evaluator.py \
    --ref_batch ${REF_BATCH} \
    --sample_batch ${SAMPLE_DIR}/depth-mu-2-threshold-0.5-0550000-base-cfg-1.0-64-SDE-100-Euler-sigma-Mean-0.04.npz \
    > eval_threshold_0.5.log 2>&1 &

# Evaluate threshold 0.75 on GPU 4
CUDA_VISIBLE_DEVICES=0 nohup python evaluator.py \
    --ref_batch ${REF_BATCH} \
    --sample_batch ${SAMPLE_DIR}/depth-mu-2-threshold-0.75-0550000-base-cfg-1.0-64-SDE-100-Euler-sigma-Mean-0.04.npz \
    > eval_threshold_0.75.log 2>&1 &

# Evaluate threshold 1.0 on GPU 5
CUDA_VISIBLE_DEVICES=1 nohup python evaluator.py \
    --ref_batch ${REF_BATCH} \
    --sample_batch ${SAMPLE_DIR}/depth-mu-2-threshold-1.0-0550000-base-cfg-1.0-64-SDE-100-Euler-sigma-Mean-0.04.npz \
    > eval_threshold_1.0.log 2>&1 &

# Wait for all background jobs to complete
echo "All evaluation tasks started. Waiting for completion..."
wait

echo "All evaluation tasks completed!"
echo ""
echo "Results saved in:"
echo "  - eval_threshold_0.0.log"
echo "  - eval_threshold_0.15.log"
echo "  - eval_threshold_0.25.log"
echo "  - eval_threshold_0.5.log"
echo "  - eval_threshold_0.75.log"
echo "  - eval_threshold_1.0.log"