diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-25 23:32:50 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-25 23:32:50 +0200 |
| commit | 9426cc794d8c28a65bbbf5ae5466a0a343078558 (patch) | |
| tree | 44e31b0a7c58597d603ac29a693462aae4b6e9b0 /training/configs/vqvae.yaml | |
| parent | 4e60c836fb710baceba570c28c06437db3ad5c9b (diff) | |
Efficient net and non working transformer model.
Diffstat (limited to 'training/configs/vqvae.yaml')
| -rw-r--r-- | training/configs/vqvae.yaml | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/training/configs/vqvae.yaml b/training/configs/vqvae.yaml index a7acb3a..13d7c97 100644 --- a/training/configs/vqvae.yaml +++ b/training/configs/vqvae.yaml @@ -5,12 +5,12 @@ network: type: VQVAE args: in_channels: 1 - channels: [32, 64, 64] - kernel_sizes: [4, 4, 4] - strides: [2, 2, 2] + channels: [32, 64, 64, 96, 96] + kernel_sizes: [4, 4, 4, 4, 4] + strides: [2, 2, 2, 2, 2] num_residual_layers: 2 - embedding_dim: 128 - num_embeddings: 512 + embedding_dim: 512 + num_embeddings: 1024 upsampling: null beta: 0.25 activation: leaky_relu |