From b44de0e11281c723ec426f8bec8ca0897ecfe3ff Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 21 Nov 2021 21:34:53 +0100
Subject: Remove VQVAE stuff, did not work...

---
 training/conf/network/decoder/vae_decoder.yaml |  9 ---------
 training/conf/network/encoder/vae_encoder.yaml |  9 ---------
 training/conf/network/quantizer.yaml           | 12 ------------
 training/conf/network/vqvae.yaml               | 18 ------------------
 4 files changed, 48 deletions(-)
 delete mode 100644 training/conf/network/decoder/vae_decoder.yaml
 delete mode 100644 training/conf/network/encoder/vae_encoder.yaml
 delete mode 100644 training/conf/network/quantizer.yaml
 delete mode 100644 training/conf/network/vqvae.yaml

(limited to 'training/conf/network')

diff --git a/training/conf/network/decoder/vae_decoder.yaml b/training/conf/network/decoder/vae_decoder.yaml
deleted file mode 100644
index aed5733..0000000
--- a/training/conf/network/decoder/vae_decoder.yaml
+++ /dev/null
@@ -1,9 +0,0 @@
-_target_: text_recognizer.networks.vqvae.decoder.Decoder
-out_channels: 1 
-hidden_dim: 32
-channels_multipliers: [4, 2, 1]
-dropout_rate: 0.0
-activation: mish
-use_norm: true
-num_residuals: 4
-residual_channels: 32
diff --git a/training/conf/network/encoder/vae_encoder.yaml b/training/conf/network/encoder/vae_encoder.yaml
deleted file mode 100644
index 5d39bf7..0000000
--- a/training/conf/network/encoder/vae_encoder.yaml
+++ /dev/null
@@ -1,9 +0,0 @@
-_target_: text_recognizer.networks.vqvae.encoder.Encoder
-in_channels: 1 
-hidden_dim: 32
-channels_multipliers: [1, 2, 4]
-dropout_rate: 0.0
-activation: mish
-use_norm: true
-num_residuals: 4
-residual_channels: 32
diff --git a/training/conf/network/quantizer.yaml b/training/conf/network/quantizer.yaml
deleted file mode 100644
index 827a247..0000000
--- a/training/conf/network/quantizer.yaml
+++ /dev/null
@@ -1,12 +0,0 @@
-_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
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
-- 
cgit v1.2.3-70-g09d2