gan_cifar10/config-2023-06-19-06-41-43.yaml

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