training/conf/network/vqvae.yaml


1
2
3
4
5
6
7
8
9
10
11
12
13

type: VQVAE
args:
  in_channels: 1
  channels: [64, 96]
  kernel_sizes: [4, 4]
  strides: [2, 2]
  num_residual_layers: 2 
  embedding_dim: 64
  num_embeddings: 256
  upsampling: null
  beta: 0.25
  activation: leaky_relu
  dropout_rate: 0.2