face_train.yaml 528 Bytes
Newer Older
0x3f3f3f3fun's avatar
0x3f3f3f3fun committed
1
2
3
4
5
6
dataset:
  target: dataset.codeformer.CodeformerDataset
  params:
    # Path to the file list.
    file_list: 
    out_size: 512
0x3f3f3f3fun's avatar
0x3f3f3f3fun committed
7
    crop_type: none
0x3f3f3f3fun's avatar
0x3f3f3f3fun committed
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
    use_hflip: False
    
    blur_kernel_size: 41
    kernel_list: ['iso', 'aniso']
    kernel_prob: [0.5, 0.5]
    blur_sigma: [0.1, 10]
    downsample_range: [0.8, 8]
    noise_range: [0, 20]
    jpeg_range: [60, 100]

data_loader:
  batch_size: 16
  shuffle: true
  num_workers: 16
  drop_last: true

batch_transform:
  target: dataset.batch_transform.IdentityBatchTransform