common:
  name: sp2pII-phase4
  model: style_based_pix2pixII
  gpu_ids:
  - 0

dataset:
  batch_size: 8
  crop_size: 512
  drop_last: true
  load_size: 512
  preprocess:
  - resize
  unpaired_trainA_folder: "/share/group_machongyang/project/hand_drawn/data/dataset/0909/trainPA/" # source domain folder path(FFHQ) 
  unpaired_trainB_folder: "/share/group_machongyang/project/hand_drawn/data/dataset/0909/trainPA/" # source domain folder path(AAHQ) 
model:
  ngf: 64
testing:
  crop_size: 512
  image_format: png
  load_size: 512
  num_test: 100000
  preprocess:
  - resize
  results_dir: ./results/sp2pII
  visual_names:
  - fake_B
  which_epoch: latest
training:
  beta1: 0.9
  epoch_as_iter: true
  lambda_Feat: 4.0
  lambda_GAN: 1.0
  lambda_L1: 1.0
  lambda_ST: 0.5
  lambda_CLIP: 1.0 # 这个参数需要调整
  lambda_PROJ: 100.0 # 这个参数需要调整(仅使用image prompt情况)
  ema: 0.99 # 1-1/n
  text_prompt: "not existed"
  image_prompt: "" # 如果这个文件存在就用image prompt, 否则用text prompt
  lr: 0.0002 # 这个参数需要调整, 大概1e-5 ~ 2e-4之间
  lr_policy: linear
  n_epochs: 200 # 这个一般500 iter就够了
  n_epochs_decay: 10
  phase: 4
  pretrained_model: pretrained_models/phase3_pretrain_10000.pth
  print_freq: 50
  save_epoch_freq: 200
  style_mixing_prob: 0.5