File size: 686 Bytes
087921f |
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 |
# basic setting
is_train: True
worker: 1
# train
# basic
batch_size: 3
shuffle: true
max_epoch: 500
epoch_start: 1 # the starting epoch count
# optimizer
beta1: 0.5
beta2: 0.999
weight_decay: 0.0001
lr: 0.0001
# print
visual_img_freq: 8000
print_loss_freq: 1000
save_epoch_freq: 20
# test
test_batch_size: 1 # must be one
results_root: 'results/'
test_freq: 5
# dataset
input_nc: 3
mask_nc: 1
output_nc: 3
crop_size: 256
crop: True
flip: True
# generator
ngf: 32
G_norm_type: in
# discriminator
ndf: 64
D_norm_type: batch
gan_mode: hinge
# loss
no_ganFeat_loss: False
no_vgg_loss: False
lambda_L1: 1
lambda_feat: 10
lambda_vgg: 10
lambda_gan: 1
# SEAN
style_length: 128
|