summaryrefslogtreecommitdiff
path: root/training/conf/network/vqvae.yaml
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-21 21:34:53 +0100
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-21 21:34:53 +0100
commitb44de0e11281c723ec426f8bec8ca0897ecfe3ff (patch)
tree998841a3a681d3dedfbe8470c1b8544b4dcbe7a2 /training/conf/network/vqvae.yaml
parent3b2fb0fd977a6aff4dcf88e1a0f99faac51e05b1 (diff)
Remove VQVAE stuff, did not work...
Diffstat (limited to 'training/conf/network/vqvae.yaml')
-rw-r--r--training/conf/network/vqvae.yaml18
1 files changed, 0 insertions, 18 deletions
diff --git a/training/conf/network/vqvae.yaml b/training/conf/network/vqvae.yaml
deleted file mode 100644
index 22f786f..0000000
--- a/training/conf/network/vqvae.yaml
+++ /dev/null
@@ -1,18 +0,0 @@
-defaults:
- - encoder: vae_encoder
- - decoder: vae_decoder
-
-_target_: text_recognizer.networks.vqvae.vqvae.VQVAE
-quantizer:
- _target_: text_recognizer.networks.quantizer.quantizer.VectorQuantizer
- input_dim: 128
- codebook:
- _target_: text_recognizer.networks.quantizer.codebook.CosineSimilarityCodebook
- dim: 8
- codebook_size: 512
- kmeans_init: true
- kmeans_iters: 10
- decay: 0.8
- eps: 1.0e-5
- threshold_dead: 2
- commitment: 1.0