forked from NVlabs/RED-diff
-
Notifications
You must be signed in to change notification settings - Fork 0
/
sample_test.sh
131 lines (78 loc) · 12.3 KB
/
sample_test.sh
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
# Copyright (c) 2023, NVIDIA CORPORATION. All rights reserved
samples_root=/home/mmardani/research/stable-diffusion-sampling-gitlab/reddiff/_exp
save_deg=True
save_ori=True
overwrite=True
smoke_test=1
batch_size=1
num_steps=10
#noisy inpaint + reddiff + adam
python main.py exp.overwrite=True algo=reddiff exp.seed=1 algo.sigma_x0=0.0 algo.awd=True algo.deg=in2_20ff algo.lr=0.25 exp.num_steps=$num_steps algo.sigma_y=0.1 loader.batch_size=$batch_size loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root exp.save_evolution=True algo.grad_term_weight=1.0
#noisy inpaint + dps
#tune eta 0.0 0.5 1.0
#grad_term_weight 0.1 0.25 0.5 1.0 2.0
#python main.py exp.overwrite=True algo=dps algo.deg=in2_20ff algo.eta=0.5 exp.num_steps=$num_steps algo.sigma_y=0.1 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=0.5
#noisy inpaint + pgdm
#python main.py exp.overwrite=True algo=pgdm algo.awd=True algo.deg=in2_20ff algo.eta=1.0 exp.num_steps=100 algo.sigma_y=0.1 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root
#inpaint + ddrm
#python main.py exp.overwrite=True algo=ddrm algo.deg=in2_20ff algo.eta=1.0 exp.num_steps=20 algo.sigma_y=0.1 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root
# #JPEG restoration
# #JPEG + reddiff + adam
# python main.py exp.overwrite=True algo=reddiff algo.awd=True algo.deg=jpeg20 algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=0.25 exp.save_evolution=True algo.lr=0.5 exp.start_step=1000 exp.end_step=0
#Nonlinear HDR
#HDR + reddiff + adam
#python main.py exp.overwrite=True algo=reddiff algo.awd=True algo.deg=hdr algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=0.25 exp.save_evolution=True algo.lr=0.5 exp.start_step=1000 exp.end_step=0
#HDR + dps
#python main.py exp.overwrite=True algo=dps algo.awd=True algo.deg=hdr algo.eta=0.5 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=0.1
#deblur + pgdm + pgdm
#python main.py exp.overwrite=True algo=pgdm algo.awd=True algo.deg=hdr algo.eta=1.0 exp.num_steps=100 algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root
#Nonlinear phase retrieval
#phase retrieval + reddiff + adam
#python main.py exp.overwrite=True algo=reddiff algo.awd=True algo.deg=phase_retrieval algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=0.25 exp.save_evolution=True algo.lr=0.5 exp.start_step=1000 exp.end_step=0
#phase retrieval + dps
#python main.py exp.overwrite=True algo=dps algo.awd=True algo.deg=phase_retrieval algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=0.4
#Nonlinear deblurring
#deblur + reddiff + adam
#python main.py exp.overwrite=True algo=reddiff algo.awd=True algo.deg=deblur_nl algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=0.25 exp.save_evolution=False algo.lr=0.5 exp.start_step=1000 exp.end_step=0
#python main.py exp.overwrite=True algo=dps algo.awd=True algo.deg=deblur_nl algo.eta=0.5 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=0.05
#Gauss deblurring
#deblur + reddiff + adam
#python main.py exp.overwrite=True algo=reddiff algo.awd=True algo.deg=deblur_gauss algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=0.25 exp.save_evolution=True algo.lr=0.5 exp.start_step=1000 exp.end_step=0
#deblur + pgdm + pgdm
#python main.py exp.overwrite=True algo=pgdm algo.awd=True algo.deg=deblur_gauss algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root
#deblur + dps
#python main.py exp.overwrite=True algo=dps algo.awd=True algo.deg=deblur_gauss algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=2.0
#deblur + ddrm
#python main.py exp.overwrite=True algo=ddrm algo.deg=deblur_gauss algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root -cn imagenet256_cond #model.ckpt=imagenet256_cond
#SR
#sr + reddiff + adam + parallel
#python main.py exp.overwrite=True algo=reddiff_parallel algo.awd=True algo.deg=sr4 algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root
#sr + reddiff + adam
#python main.py exp.overwrite=True algo=reddiff algo.awd=True algo.deg=sr4 algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root
#sr + pgdm + pgdm
#python main.py exp.overwrite=True algo=pgdm algo.awd=True algo.deg=sr4 algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root
#sr + dps
#choose eta
#grad_term_weight 0.1 0.25 0.5 1.0 2.0
#python main.py exp.overwrite=$overwrite algo=dps algo.deg=sr4 algo.eta=0.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=1.0
#sr + ddrm
#python main.py exp.overwrite=True algo=ddrm algo.deg=sr4 algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root -cn imagenet256_cond #model.ckpt=imagenet256_cond
#INPAINT
#inpaint + reddiff + adam + parallel
#python main.py exp.overwrite=True algo=reddiff_parallel algo.grad_term_weight=0.5 algo.lr=0.25 algo.awd=True algo.deg=in2_20ff algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root
#inpaint + reddiff + adam
#python main.py exp.overwrite=True algo=reddiff exp.seed=1 algo.sigma_x0=0.0001 algo.lr=0.25 algo.awd=True algo.deg=in2_20ff algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root
#inpaint + pgdm
#python main.py exp.overwrite=True algo=pgdm algo.awd=True algo.deg=in2_20ff algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root -cn imagenet256_cond #model.ckpt=imagenet256_cond
#inpaint + mcg
#gives nans
#grad_term_weight needs to be tuned properly
#choose eta
#grad_term_weight ??
#python main.py exp.overwrite=True algo=mcg algo.deg=in2_20ff algo.eta=0.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root
#inpaint + dps
#choose eta
#grad_term_weight 0.1 0.25 0.5 1.0 2.0
#python main.py exp.overwrite=True algo=dps algo.deg=in2_20ff algo.eta=0.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root algo.grad_term_weight=0.5
#inpaint + ddrm
#python main.py exp.overwrite=True algo=ddrm algo.deg=in2_20ff algo.eta=1.0 exp.num_steps=$num_steps algo.sigma_y=0.0 loader.batch_size=$batch_size exp.seed=3 loader=imagenet256_ddrmpp dist.num_processes_per_node=1 exp.name=debug exp.save_ori=$save_ori exp.save_deg=$save_deg exp.smoke_test=$smoke_test exp.samples_root=$samples_root -cn imagenet256_cond #model.ckpt=imagenet256_cond