common:
  name: "sp2pII-phase3"
  model: "style_based_pix2pixII"
  gpu_ids: [0]
  option_group:
    - gpu_ids: [0]
    - gpu_ids: [1]
    - gpu_ids: [2]

model:
  ngf: 64

dataset:
  unpaired_trainA_folder: # source domain folder path(FFHQ) 
  unpaired_trainB_folder: # target domain folder path(AAHQ) 
  preprocess: ["resize"]
  batch_size: 8
  crop_size: 512
  drop_last: true
  load_size: 512

training:
  epoch_as_iter: true
  n_epochs: 100000 # 这个收敛很快，10w iter就差不多了
  n_epochs_decay: 10
  print_freq: 1000
  phase: 3
  pretrained_model: "pretrained_models/phase2_pretrain_90000.pth"
  save_epoch_freq: 5000
  style_mixing_prob: 0.5
  lambda_GAN: 1.0
  lambda_ST: 1.0
  lambda_L1: 1.0
  option_group:
    - lambda_Feat: 4.0
    - lambda_Feat: 2.0
    - lambda_Feat: 1.0
  lr: 0.0002
  lr_policy: "linear"
  beta1: 0.9

testing:
  num_test: 100000
  preprocess: ["resize"]
  load_size: 512
  crop_size: 512
  results_dir: "./results/sp2pII"
  visual_names: ["fake_B"]
  image_format: "png"
  which_epoch: "latest"