File size: 935 Bytes
4870d91
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
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
    with_sn: 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.hinge_loss.HingeLoss
    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