Spaces:
Sleeping
Sleeping
| 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 | |