|
scale: 4 |
|
num_gpu: 1 |
|
manual_seed: 0 |
|
is_train: True |
|
dist: False |
|
|
|
|
|
|
|
l1_gt_usm: True |
|
percep_gt_usm: True |
|
gan_gt_usm: False |
|
|
|
|
|
resize_prob: [0.2, 0.7, 0.1] |
|
resize_range: [0.15, 1.5] |
|
gaussian_noise_prob: 1 |
|
noise_range: [1, 30] |
|
poisson_scale_range: [0.05, 3] |
|
gray_noise_prob: 1 |
|
jpeg_range: [30, 95] |
|
|
|
|
|
second_blur_prob: 1 |
|
resize_prob2: [0.3, 0.4, 0.3] |
|
resize_range2: [0.3, 1.2] |
|
gaussian_noise_prob2: 1 |
|
noise_range2: [1, 25] |
|
poisson_scale_range2: [0.05, 2.5] |
|
gray_noise_prob2: 1 |
|
jpeg_range2: [30, 95] |
|
|
|
gt_size: 32 |
|
queue_size: 1 |
|
|
|
|
|
network_g: |
|
type: RRDBNet |
|
num_in_ch: 3 |
|
num_out_ch: 3 |
|
num_feat: 4 |
|
num_block: 1 |
|
num_grow_ch: 2 |
|
|
|
network_d: |
|
type: UNetDiscriminatorSN |
|
num_in_ch: 3 |
|
num_feat: 2 |
|
skip_connection: True |
|
|
|
|
|
path: |
|
pretrain_network_g: ~ |
|
param_key_g: params_ema |
|
strict_load_g: true |
|
resume_state: ~ |
|
|
|
|
|
train: |
|
ema_decay: 0.999 |
|
optim_g: |
|
type: Adam |
|
lr: !!float 1e-4 |
|
weight_decay: 0 |
|
betas: [0.9, 0.99] |
|
optim_d: |
|
type: Adam |
|
lr: !!float 1e-4 |
|
weight_decay: 0 |
|
betas: [0.9, 0.99] |
|
|
|
scheduler: |
|
type: MultiStepLR |
|
milestones: [400000] |
|
gamma: 0.5 |
|
|
|
total_iter: 400000 |
|
warmup_iter: -1 |
|
|
|
|
|
pixel_opt: |
|
type: L1Loss |
|
loss_weight: 1.0 |
|
reduction: mean |
|
|
|
perceptual_opt: |
|
type: PerceptualLoss |
|
layer_weights: |
|
|
|
'conv1_2': 0.1 |
|
'conv2_2': 0.1 |
|
'conv3_4': 1 |
|
'conv4_4': 1 |
|
'conv5_4': 1 |
|
vgg_type: vgg19 |
|
use_input_norm: true |
|
perceptual_weight: !!float 1.0 |
|
style_weight: 0 |
|
range_norm: false |
|
criterion: l1 |
|
|
|
gan_opt: |
|
type: GANLoss |
|
gan_type: vanilla |
|
real_label_val: 1.0 |
|
fake_label_val: 0.0 |
|
loss_weight: !!float 1e-1 |
|
|
|
net_d_iters: 1 |
|
net_d_init_iters: 0 |
|
|
|
|
|
|
|
val: |
|
val_freq: !!float 5e3 |
|
save_img: False |
|
|