-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtune_pgdm.bash
More file actions
28 lines (21 loc) · 1.3 KB
/
tune_pgdm.bash
File metadata and controls
28 lines (21 loc) · 1.3 KB
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
#!/bin/bash
# Tune hyperparameters for PGDM
# only one hyperparameter: starting_time
# nfe_list=(20 40 100 200)
noise_levels=(0.0 0.05 0.1 1.0)
# noise_levels=(0.05)
starting_times=(0.0 0.1 0.2 0.3 0.4 0.5 0.6)
# starting_times=(0.0 0.1)
max_samp=100
for noise_lv in ${noise_levels[@]}; do
echo "Tuning hyperparameters for PGDM with noise level ${noise_lv}"
for starting_time in ${starting_times[@]}; do
echo "Sampling with starting time ${starting_time}"
python run_sampling.py --config configs/pgdm/afhq/inpaint_pixel.py --max_num_samples ${max_samp} --compute_recon_metrics --noise_level ${noise_lv} --starting_time ${starting_time}
python run_sampling.py --config configs/pgdm/afhq/inpaint_box.py --max_num_samples ${max_samp} --compute_recon_metrics --noise_level ${noise_lv} --starting_time ${starting_time}
python run_sampling.py --config configs/pgdm/afhq/deblur.py --max_num_samples ${max_samp} --compute_recon_metrics --noise_level ${noise_lv} --starting_time ${starting_time}
python run_sampling.py --config configs/pgdm/afhq/super_res.py --max_num_samples ${max_samp} --compute_recon_metrics --noise_level ${noise_lv} --starting_time ${starting_time}
done
done
# find best hyperparameters
python evaluation/find_hyp.py --method pgdm --data_name afhq