summaryrefslogtreecommitdiff
path: root/training/conf/network
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-08-30 21:18:32 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-08-30 21:18:32 +0200
commitca7e91a2a97dc54b8201cc7d0b1ce3eb4a3ee1bd (patch)
tree69710c10f7f683237cd3c97ff8e2407818587e6f /training/conf/network
parent2f1bb639fd5bb6b510af85fb597e9322abc17bc0 (diff)
Update dependencies, update to vqvae params
Diffstat (limited to 'training/conf/network')
-rw-r--r--training/conf/network/decoder/vae_decoder.yaml4
-rw-r--r--training/conf/network/encoder/vae_encoder.yaml4
-rw-r--r--training/conf/network/vqvae.yaml4
3 files changed, 6 insertions, 6 deletions
diff --git a/training/conf/network/decoder/vae_decoder.yaml b/training/conf/network/decoder/vae_decoder.yaml
index 7558ff0..60cdcf1 100644
--- a/training/conf/network/decoder/vae_decoder.yaml
+++ b/training/conf/network/decoder/vae_decoder.yaml
@@ -1,5 +1,5 @@
_target_: text_recognizer.networks.vqvae.decoder.Decoder
out_channels: 1
-hidden_dim: 32
-channels_multipliers: [4, 4, 2, 1]
+hidden_dim: 64
+channels_multipliers: [8, 4, 2, 1]
dropout_rate: 0.0
diff --git a/training/conf/network/encoder/vae_encoder.yaml b/training/conf/network/encoder/vae_encoder.yaml
index b32f425..73529fc 100644
--- a/training/conf/network/encoder/vae_encoder.yaml
+++ b/training/conf/network/encoder/vae_encoder.yaml
@@ -1,5 +1,5 @@
_target_: text_recognizer.networks.vqvae.encoder.Encoder
in_channels: 1
-hidden_dim: 32
-channels_multipliers: [1, 2, 4, 4]
+hidden_dim: 64
+channels_multipliers: [1, 2, 4, 8]
dropout_rate: 0.0
diff --git a/training/conf/network/vqvae.yaml b/training/conf/network/vqvae.yaml
index 936e575..70d27d7 100644
--- a/training/conf/network/vqvae.yaml
+++ b/training/conf/network/vqvae.yaml
@@ -3,7 +3,7 @@ defaults:
- decoder: vae_decoder
_target_: text_recognizer.networks.vqvae.vqvae.VQVAE
-hidden_dim: 128
+hidden_dim: 512
embedding_dim: 64
-num_embeddings: 2048
+num_embeddings: 4096
decay: 0.99