BasicSR

1
# general settings
2
name: 052_ESRGAN_x4_f64b23_DIV2K_400k_B16G1_051pretrain_wandb
3
model_type: ESRGANModel
4
scale: 4
5
num_gpu: 1  # set num_gpu: 0 for cpu mode
6
manual_seed: 0
7

8
# dataset and data loader settings
9
datasets:
10
  train:
11
    name: DIV2K
12
    type: PairedImageDataset
13
    dataroot_gt: datasets/DIV2K/DIV2K_train_HR_sub
14
    dataroot_lq: datasets/DIV2K/DIV2K_train_LR_bicubic/X4_sub
15
    # (for lmdb)
16
    # dataroot_gt: datasets/DIV2K/DIV2K_train_HR_sub.lmdb
17
    # dataroot_lq: datasets/DIV2K/DIV2K_train_LR_bicubic_X4_sub.lmdb
18
    filename_tmpl: '{}'
19
    io_backend:
20
      type: disk
21
      # (for lmdb)
22
      # type: lmdb
23

24
    gt_size: 128
25
    use_hflip: true
26
    use_rot: true
27

28
    # data loader
29
    num_worker_per_gpu: 6
30
    batch_size_per_gpu: 16
31
    dataset_enlarge_ratio: 100
32
    prefetch_mode: ~
33

34
  val:
35
    name: Set14
36
    type: PairedImageDataset
37
    dataroot_gt: datasets/Set14/GTmod12
38
    dataroot_lq: datasets/Set14/LRbicx4
39
    io_backend:
40
      type: disk
41

42
# network structures
43
network_g:
44
  type: RRDBNet
45
  num_in_ch: 3
46
  num_out_ch: 3
47
  num_feat: 64
48
  num_block: 23
49

50
network_d:
51
  type: VGGStyleDiscriminator
52
  num_in_ch: 3
53
  num_feat: 64
54

55
# path
56
path:
57
  pretrain_network_g: experiments/051_RRDBNet_PSNR_x4_f64b23_DIV2K_1000k_B16G1_wandb/models/net_g_1000000.pth
58
  strict_load_g: true
59
  resume_state: ~
60

61
# training settings
62
train:
63
  ema_decay: 0.999
64
  optim_g:
65
    type: Adam
66
    lr: !!float 1e-4
67
    weight_decay: 0
68
    betas: [0.9, 0.99]
69
  optim_d:
70
    type: Adam
71
    lr: !!float 1e-4
72
    weight_decay: 0
73
    betas: [0.9, 0.99]
74

75
  scheduler:
76
    type: MultiStepLR
77
    milestones: [50000, 100000, 200000, 300000]
78
    gamma: 0.5
79

80
  total_iter: 400000
81
  warmup_iter: -1  # no warm up
82

83
  # losses
84
  pixel_opt:
85
    type: L1Loss
86
    loss_weight: !!float 1e-2
87
    reduction: mean
88
  perceptual_opt:
89
    type: PerceptualLoss
90
    layer_weights:
91
      'conv5_4': 1  # before relu
92
    vgg_type: vgg19
93
    use_input_norm: true
94
    range_norm: false
95
    perceptual_weight: 1.0
96
    style_weight: 0
97
    criterion: l1
98
  gan_opt:
99
    type: GANLoss
100
    gan_type: vanilla
101
    real_label_val: 1.0
102
    fake_label_val: 0.0
103
    loss_weight: !!float 5e-3
104

105
  net_d_iters: 1
106
  net_d_init_iters: 0
107

108
# validation settings
109
val:
110
  val_freq: !!float 5e3
111
  save_img: true
112

113
  metrics:
114
    psnr: # metric name, can be arbitrary
115
      type: calculate_psnr
116
      crop_border: 4
117
      test_y_channel: false
118

119
# logging settings
120
logger:
121
  print_freq: 100
122
  save_checkpoint_freq: !!float 5e3
123
  use_tb_logger: true
124
  wandb:
125
    project: ~
126
    resume_id: ~
127

128
# dist training settings
129
dist_params:
130
  backend: nccl
131
  port: 29500
132