summaryrefslogtreecommitdiff
path: root/training/conf/network/vqvae.yaml
blob: 8c30bbdb5095f4e7f00397e3d8bbf708c5db0b8c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# @package _group_
type: VQVAE
args:
    in_channels: 1
    channels: [32, 64, 64]
    kernel_sizes: [4, 4, 4]
    strides: [2, 2, 2]
    num_residual_layers: 2 
    embedding_dim: 64
    num_embeddings: 256
    upsampling: null
    beta: 0.25
    activation: leaky_relu
    dropout_rate: 0.2