Hugging Face's logo

NorskRegnesentralSTI
/
DiffFuSR

Image-to-Image
Model card Files
xet
Community
2
msarmad commited on
Commit
17d6d17
·
verified ·
1 Parent(s): 56a62ed

Create hparams.yml

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