Spaces:
Sleeping
Sleeping
| common: | |
| name: "sp2pII-phase2" | |
| 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: 300000 | |
| n_epochs_decay: 10 | |
| print_freq: 1000 | |
| phase: 2 | |
| pretrained_model: "pretrained_models/ffhq_pretrain_res512_200000.pt" # phase1 model | |
| save_epoch_freq: 5000 | |
| style_mixing_prob: 0.5 | |
| lambda_GAN: 1.0 | |
| lambda_ST: 0.5 # 这个参数可以调整 | |
| option_group: | |
| - data_aug_prob: 0.0 | |
| - data_aug_prob: 0.1 | |
| - data_aug_prob: 0.2 | |
| lr: 0.001 | |
| lr_policy: "linear" | |
| beta1: 0.1 | |
| 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" | |