|
|
|
|
|
|
|
|
|
|
|
distributed: False |
|
image_to_tensorboard: True |
|
snapshot_save_iter: 40000 |
|
snapshot_save_epoch: 20 |
|
snapshot_save_start_iter: 20000 |
|
snapshot_save_start_epoch: 10 |
|
image_save_iter: 1000 |
|
max_epoch: 200 |
|
logging_iter: 100 |
|
results_dir: ./eval_results |
|
|
|
gen_optimizer: |
|
type: adam |
|
lr: 0.0001 |
|
adam_beta1: 0.5 |
|
adam_beta2: 0.999 |
|
lr_policy: |
|
iteration_mode: True |
|
type: step |
|
step_size: 300000 |
|
gamma: 0.2 |
|
|
|
trainer: |
|
type: trainers.face_trainer::FaceTrainer |
|
pretrain_warp_iteration: 200000 |
|
loss_weight: |
|
weight_perceptual_warp: 2.5 |
|
weight_perceptual_final: 4 |
|
vgg_param_warp: |
|
network: vgg19 |
|
layers: ['relu_1_1', 'relu_2_1', 'relu_3_1', 'relu_4_1', 'relu_5_1'] |
|
use_style_loss: False |
|
num_scales: 4 |
|
vgg_param_final: |
|
network: vgg19 |
|
layers: ['relu_1_1', 'relu_2_1', 'relu_3_1', 'relu_4_1', 'relu_5_1'] |
|
use_style_loss: True |
|
num_scales: 4 |
|
style_to_perceptual: 250 |
|
init: |
|
type: 'normal' |
|
gain: 0.02 |
|
gen: |
|
type: generators.face_model::FaceGenerator |
|
param: |
|
mapping_net: |
|
coeff_nc: 73 |
|
descriptor_nc: 256 |
|
layer: 3 |
|
warpping_net: |
|
encoder_layer: 5 |
|
decoder_layer: 3 |
|
base_nc: 32 |
|
editing_net: |
|
layer: 3 |
|
num_res_blocks: 2 |
|
base_nc: 64 |
|
common: |
|
image_nc: 3 |
|
descriptor_nc: 256 |
|
max_nc: 256 |
|
use_spect: False |
|
|
|
|
|
|
|
data: |
|
type: data.vox_dataset::VoxDataset |
|
path: ./dataset/vox_lmdb |
|
resolution: 256 |
|
semantic_radius: 13 |
|
train: |
|
batch_size: 5 |
|
distributed: True |
|
val: |
|
batch_size: 8 |
|
distributed: True |
|
|
|
|
|
|