File size: 1,831 Bytes
9ab094a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
# How often do you want to log the training stats.
# network_list:
# gen: gen_optimizer
# dis: dis_optimizer
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 options.
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
|