data:
  dataroot: /amax/xyf/data/ImageNet/
  img_size: 256
  name: ImageNet
dataloader:
  micro_batch: 0
  num_workers: 4
  pin_memory: true
  prefetch_factor: 2
mask:
  dir_invert_color: true
  dir_path: /amax/xyf/data/NVIDIAIrregularMaskDataset/train/
  mask_type: dir
  noise_type: real
  real_dataset:
  - dataroot: /amax/xyf/data/Places365/
    img_size: 256
    name: Places365
  smooth_kernel_size: 13
  smooth_sigma: 3
  smooth_type: gaussian
model:
  mpn:
    base_n_channels: 64
    neck_n_channels: 128
  rin:
    base_n_channels: 32
    neck_n_channels: 128
seed: 1234
train:
  batch_size: 4
  coef_adv: 0.001
  coef_idmrf: 0.001
  coef_mpn: 2.0
  coef_rec: 1.4
  coef_semantic: 0.0001
  eval_freq: 1000
  n_steps: 150000
  optim_disc:
    betas:
    - 0.5
    - 0.9
    lr: 1.0e-05
  optim_mpn:
    betas:
    - 0.5
    - 0.9
    lr: 1.0e-05
  optim_rin:
    betas:
    - 0.5
    - 0.9
    lr: 1.0e-05
  pretrained: ./runs/separate_imagenet_maskdir_realnoise/ckpt/step149999/model.pt
  print_freq: 100
  resume: null
  sample_freq: 1000
  save_freq: 5000