mvq / config / 64-VQGAN-CB.yaml
64-VQGAN-CB.yaml
Raw
---
data:
  name: celeba64
  in_shape: [64, 64]
  in_channels: 3
model:
  backbone: vq2
  vq:
    ema_update: true
    random_restart: false
    tiled: false
  mhd:
    use: false
  gan:
    use: true
train:
  epochs:
    vae: 251