File size: 741 Bytes
f2cb7e8 |
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 |
seed: 8888
data:
target: datasets.celeba.CelebA
params:
root: ~/data/CelebA/
img_size: 64
split: train
dataloader:
num_workers: 4
pin_memory: true
prefetch_factor: 2
encoder:
target: models.simple_cnn.Encoder
params:
z_dim: 128
in_dim: 3
dim: 64
dim_mults:
- 1
- 2
- 4
- 8
with_bn: true
decoder:
target: models.simple_cnn.Decoder
params:
z_dim: 128
dim: 64
dim_mults:
- 8
- 4
- 2
- 1
out_dim: 3
with_bn: true
with_tanh: true
train:
n_steps: 150000
batch_size: 256
resume: null
print_freq: 100
sample_freq: 1000
save_freq: 10000
n_samples: 64
coef_kl: 20
optim:
target: torch.optim.Adam
params:
lr: 0.02
|