Hugging Face's logo

NorskRegnesentralSTI
/
DiffFuSR

Model card Files
xet
Community
1
msarmad commited on
Commit
9086a95
·
verified ·
1 Parent(s): 7d24f19

Create hparams.yaml

Browse files
Files changed (1) hide show
  1. logs/blindsrsnf_aniso_naip_degraded_harm_large/version_1/hparams.yaml +211 -0
logs/blindsrsnf_aniso_naip_degraded_harm_large/version_1/hparams.yaml ADDED
@@ -0,0 +1,211 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ data_module: !!python/object/new:easydict.EasyDict
2
+ dictitems:
3
+ args: &id003 !!python/object/new:easydict.EasyDict
4
+ dictitems:
5
+ channels: 4
6
+ multi_spectral: true
7
+ num_workers: 4
8
+ rgb_range: 1
9
+ scale: 4
10
+ train: &id001 !!python/object/new:easydict.EasyDict
11
+ dictitems:
12
+ batch_size: 8
13
+ data_cache: memory
14
+ data_length: 32768
15
+ data_path: load/naip/synthetic_naip
16
+ harmonization: true
17
+ hr_img_sz: 256
18
+ state:
19
+ batch_size: 8
20
+ data_cache: memory
21
+ data_length: 32768
22
+ data_path: load/naip/synthetic_naip
23
+ harmonization: true
24
+ hr_img_sz: 256
25
+ valid: &id002 !!python/object/new:easydict.EasyDict
26
+ dictitems:
27
+ data_first_k: 1
28
+ hr_path: load/WorldStrat_png/test/HR
29
+ lr_path: load/WorldStrat_png/test/LR
30
+ state:
31
+ data_first_k: 1
32
+ hr_path: load/WorldStrat_png/test/HR
33
+ lr_path: load/WorldStrat_png/test/LR
34
+ state:
35
+ channels: 4
36
+ multi_spectral: true
37
+ num_workers: 4
38
+ rgb_range: 1
39
+ scale: 4
40
+ train: *id001
41
+ valid: *id002
42
+ name: BlindSRJIFUSGSDataModule
43
+ state:
44
+ args: *id003
45
+ name: BlindSRJIFUSGSDataModule
46
+ lit_model: !!python/object/new:easydict.EasyDict
47
+ dictitems:
48
+ args: &id016 !!python/object/new:easydict.EasyDict
49
+ dictitems:
50
+ blur_kernel: 21
51
+ blur_type: aniso_gaussian
52
+ contrast_loss_weight: 0.05
53
+ dg_encoder: DegradationEncoder
54
+ diffusion: &id009 !!python/object/new:easydict.EasyDict
55
+ dictitems:
56
+ beta_end: 0.02
57
+ beta_start: 0.0001
58
+ beta_type: cosine
59
+ model_mean_type: xstart
60
+ model_var_type: fixedlarge
61
+ n_timestep: 1000
62
+ t_encode_mode: continuous
63
+ state:
64
+ beta_end: 0.02
65
+ beta_start: 0.0001
66
+ beta_type: cosine
67
+ model_mean_type: xstart
68
+ model_var_type: fixedlarge
69
+ n_timestep: 1000
70
+ t_encode_mode: continuous
71
+ diffusion_model: &id010 !!python/object/new:easydict.EasyDict
72
+ dictitems:
73
+ args: &id006 !!python/object/new:easydict.EasyDict
74
+ dictitems:
75
+ attn_strides: &id004
76
+ - 16
77
+ channel_mults: &id005
78
+ - 1
79
+ - 2
80
+ - 2
81
+ - 4
82
+ - 8
83
+ dropout: 0
84
+ fold: 2
85
+ in_channel: 3
86
+ inner_channel: 64
87
+ out_channel: 3
88
+ res_blocks: 5
89
+ state:
90
+ attn_strides: *id004
91
+ channel_mults: *id005
92
+ dropout: 0
93
+ fold: 2
94
+ in_channel: 3
95
+ inner_channel: 64
96
+ out_channel: 3
97
+ res_blocks: 5
98
+ name: DiffUNetSR3Atom
99
+ state:
100
+ args: *id006
101
+ name: DiffUNetSR3Atom
102
+ ema: true
103
+ ema_rate: 0.999
104
+ encoder: &id011 !!python/object/new:easydict.EasyDict
105
+ dictitems:
106
+ args: &id007 !!python/object/new:easydict.EasyDict
107
+ dictitems:
108
+ num_block: 23
109
+ num_feat: 64
110
+ num_in_ch: 3
111
+ num_out_ch: 3
112
+ scale: 4
113
+ state:
114
+ num_block: 23
115
+ num_feat: 64
116
+ num_in_ch: 3
117
+ num_out_ch: 3
118
+ scale: 4
119
+ is_train: true
120
+ name: RRDBNet
121
+ state:
122
+ args: *id007
123
+ is_train: true
124
+ name: RRDBNet
125
+ lambda_max: 4.0
126
+ lambda_min: 0.2
127
+ mean: &id012
128
+ - 0.5
129
+ - 0.5
130
+ - 0.5
131
+ noise: 25
132
+ optimizer: &id013 !!python/object/new:easydict.EasyDict
133
+ dictitems:
134
+ lr: 0.0001
135
+ lr_scheduler_gamma: 0.5
136
+ lr_scheduler_milestones: &id008
137
+ - 200
138
+ - 400
139
+ - 600
140
+ - 800
141
+ state:
142
+ lr: 0.0001
143
+ lr_scheduler_gamma: 0.5
144
+ lr_scheduler_milestones: *id008
145
+ scale: 4
146
+ sig_max: 4.0
147
+ sig_min: 2.0
148
+ std: &id014
149
+ - 0.5
150
+ - 0.5
151
+ - 0.5
152
+ theta: 0.0
153
+ valid: &id015 !!python/object/new:easydict.EasyDict
154
+ dictitems:
155
+ blur_kernel: 21
156
+ blur_type: iso_gaussian
157
+ eta: 0
158
+ no_crop_border: false
159
+ noise: 0.0
160
+ scale: 4
161
+ sig: 1.2
162
+ skip: 50
163
+ test_Y: false
164
+ state:
165
+ blur_kernel: 21
166
+ blur_type: iso_gaussian
167
+ eta: 0
168
+ no_crop_border: false
169
+ noise: 0.0
170
+ scale: 4
171
+ sig: 1.2
172
+ skip: 50
173
+ test_Y: false
174
+ state:
175
+ blur_kernel: 21
176
+ blur_type: aniso_gaussian
177
+ contrast_loss_weight: 0.05
178
+ dg_encoder: DegradationEncoder
179
+ diffusion: *id009
180
+ diffusion_model: *id010
181
+ ema: true
182
+ ema_rate: 0.999
183
+ encoder: *id011
184
+ lambda_max: 4.0
185
+ lambda_min: 0.2
186
+ mean: *id012
187
+ noise: 25
188
+ optimizer: *id013
189
+ scale: 4
190
+ sig_max: 4.0
191
+ sig_min: 2.0
192
+ std: *id014
193
+ theta: 0.0
194
+ valid: *id015
195
+ name: DiffBlindSR3AtomJIFModel
196
+ state:
197
+ args: *id016
198
+ name: DiffBlindSR3AtomJIFModel
199
+ trainer: !!python/object/new:easydict.EasyDict
200
+ dictitems:
201
+ check_val_every_n_epoch: 1
202
+ max_epochs: 1000
203
+ pretrain_epochs: 0
204
+ save_period: 100
205
+ save_top_k: 2
206
+ state:
207
+ check_val_every_n_epoch: 1
208
+ max_epochs: 1000
209
+ pretrain_epochs: 0
210
+ save_period: 100
211
+ save_top_k: 2