summaryrefslogtreecommitdiff
path: root/training
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-05-02 22:27:42 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-05-02 22:27:42 +0200
commit737000da5b44276512beffc1bdf81057df43ab2c (patch)
treed44fa30079d8db0534c14b6d53e8524e05673620 /training
parent1baeae6b414f71906bd1480d3ddc393ae878bd63 (diff)
Attention layer finished
Diffstat (limited to 'training')
-rw-r--r--training/conf/network/vqvae.yaml6
1 files changed, 3 insertions, 3 deletions
diff --git a/training/conf/network/vqvae.yaml b/training/conf/network/vqvae.yaml
index 8c30bbd..288d2aa 100644
--- a/training/conf/network/vqvae.yaml
+++ b/training/conf/network/vqvae.yaml
@@ -2,9 +2,9 @@
type: VQVAE
args:
in_channels: 1
- channels: [32, 64, 64]
- kernel_sizes: [4, 4, 4]
- strides: [2, 2, 2]
+ channels: [64, 96]
+ kernel_sizes: [4, 4]
+ strides: [2, 2]
num_residual_layers: 2
embedding_dim: 64
num_embeddings: 256