File size: 1,021 Bytes
52228a8 |
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 |
seed: 5678
data:
name: FFHQ
dataroot: ~/data/FFHQ/
img_size: 256
img_channels: 3
dataloader:
num_workers: 64
pin_memory: true
prefetch_factor: 2
G:
target: models.eigengan.Generator
params:
n_basis: 6
noise_dim: 512
out_dim: 3
dim: 16
dim_mults:
- 32
- 16
- 8
- 4
- 2
- 1
with_bn: false
with_tanh: true
D:
target: models.eigengan.Discriminator
params:
in_dim: 3
dim: 16
dim_mults:
- 2
- 4
- 8
- 16
- 32
- 32
with_bn: false
train:
n_steps: 500000
batch_size: 64
resume: null
print_freq: 100
save_freq: 10000
sample_freq: 1000
n_samples: 25
d_iters: 1
ortho_reg: 1.0
ema_decay: 0.999
loss_fn:
target: losses.hinge_loss.HingeLoss
params:
lambda_r1_reg: 10.0
optim_G:
target: torch.optim.Adam
params:
lr: 0.0001
betas:
- 0.5
- 0.999
optim_D:
target: torch.optim.Adam
params:
lr: 0.0001
betas:
- 0.5
- 0.999
|