File size: 928 Bytes
d76d7ed |
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 |
seed: 5678
data:
name: CIFAR-10
dataroot: /data/CIFAR-10/
img_size: 32
img_channels: 3
dataloader:
num_workers: 4
pin_memory: true
prefetch_factor: 2
G:
target: models.simple_cnn.Generator
params:
z_dim: 100
dim: 256
dim_mults:
- 4
- 2
- 1
out_dim: 3
with_bn: true
with_tanh: true
D:
target: models.simple_cnn.Discriminator
params:
in_dim: 3
dim: 256
dim_mults:
- 1
- 2
- 4
with_bn: true
train:
n_steps: 40000
batch_size: 512
d_iters: 5
resume: null
print_freq: 100
save_freq: 10000
sample_freq: 1000
eval_freq: 1000
n_samples: 64
loss_fn:
target: losses.vanilla_gan_loss.VanillaGANLoss
params: {}
optim_G:
target: torch.optim.Adam
params:
lr: 0.0008
betas:
- 0.5
- 0.999
optim_D:
target: torch.optim.Adam
params:
lr: 0.0008
betas:
- 0.5
- 0.999
|