File size: 2,515 Bytes
0389e9b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 | #!/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"
|