facemask.yaml
417 Bytes
settings:
root_dir: "./datasets/celeba/images/"
checkpoint_path: "weights"
sample_folder: "sample"
cuda: True
lr: 0.001
batch_size: 2
num_workers: 4
step_iters: [10000, 15000, 20000]
gamma: 0.1
d_num_layers: 3
visualize_per_iter: 500
save_per_iter: 500
print_per_iter: 10
num_epochs: 100
lambda_G: 1.0
lambda_rec_1: 100.0
lambda_rec_2: 100.0
lambda_per: 10.0
img_size: 512