summaryrefslogtreecommitdiff
path: root/training/conf/network
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/network')
-rw-r--r--training/conf/network/decoder/vae_decoder.yaml9
-rw-r--r--training/conf/network/encoder/vae_encoder.yaml9
-rw-r--r--training/conf/network/quantizer.yaml12
-rw-r--r--training/conf/network/vqvae.yaml18
4 files changed, 0 insertions, 48 deletions
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