File size: 446 Bytes
59630ba | 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 | defaults:
- base_pytorch_algo
# warmup_steps and disc_start will be halved due to GAN training
warmup_steps: 10000
gradient_clip_val: 1.0
embed_dim: 4
lossconfig:
disc_start: 40001
kl_weight: 0.000001
disc_weight: 0.5
ddconfig:
double_z: true
z_channels: 4
resolution: ${dataset.resolution}
in_channels: 3
out_ch: 3
ch: 128
ch_mult: [1, 2, 4, 4]
num_res_blocks: 2
attn_resolutions: []
dropout: 0.0
ckpt_path: null |