mvq / config / 64-MVQGAN-IN.yaml
64-MVQGAN-IN.yaml
Raw
---
data:
  name: imagenet64
  in_shape: [64, 64]
  in_channels: 3
model:
  backbone: mvq
  mhd:
    use: True
    decoder:
      n_res_block: 2
  gan:
    use: True
train:
  epochs:
    vae: 651
eval:
  return_outputs: True