summaryrefslogtreecommitdiff
path: root/training/conf/network/vqvae.yaml
blob: 22eebf893f2157c8be35f46e4091d183db4c2879 (plain)
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