Back / Rectified_Noise /VP-Disp /evaluate_samples.sh
xiangzai's picture
Add files using upload-large-folder tool
0389e9b verified
#!/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"