From 2417288c9fe96264da708ce8d13ac7bc2faf83e3 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 17 Nov 2021 22:42:58 +0100
Subject: Add new quantizer

---
 training/conf/network/quantizer.yaml | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 training/conf/network/quantizer.yaml

(limited to 'training/conf/network/quantizer.yaml')

diff --git a/training/conf/network/quantizer.yaml b/training/conf/network/quantizer.yaml
new file mode 100644
index 0000000..827a247
--- /dev/null
+++ b/training/conf/network/quantizer.yaml
@@ -0,0 +1,12 @@
+_target_: text_recognizer.networks.quantizer.quantizer.VectorQuantizer
+input_dim: 192
+codebook:
+  _target_: text_recognizer.networks.quantizer.codebook.CosineSimilarityCodebook
+  dim: 16
+  codebook_size: 2048
+  kmeans_init: true
+  kmeans_iters: 10
+  decay: 0.8
+  eps: 1.0e-5
+  threshold_dead: 2
+commitment: 1.0
-- 
cgit v1.2.3-70-g09d2