google-research

1
train.model_type = "gen"
2
train.dset_name = "dsprites"
3
train.s_dim = 5
4
train.n_dim = 0
5
train.factors = "c=0,1,2,3,4"
6
mask_type = "match"
7
train.batch_size = 64
8
train.dec_lr = 1e-3
9
train.enc_lr_mul = 1
10
train.iterations = 500000
11

12
make_masks.mask_type = %mask_type
13
paired_randn.mask_type = %mask_type
14
paired_data_generator.mask_type = %mask_type
15

16
initializer.method = "keras"
17
  dense/initializer.layer = "dense"
18
  conv/initializer.layer = "conv"
19
  deconv/initializer.layer = "deconv"
20
Dense.initializer = @dense/initializer
21
Conv2d.initializer = @conv/initializer
22
ConvTranspose2d.initializer = @deconv/initializer
23

24
Encoder.width = 1
25
Encoder.spectral_norm = True
26

27
Discriminator.width = 2
28
Discriminator.share_dense = True
29
Discriminator.uncond_bias = False
30
Discriminator.mask_type = %mask_type
31

32
LabelDiscriminator.width = 1
33
LabelDiscriminator.share_dense = False
34
LabelDiscriminator.uncond_bias = False
35

36
weight_normalization.add.epsilon = 1e-3
37
weight_normalization.add.scale = False
38
batch_normalization.add.use_out_hook = True
39

40
log.debug = False
41