From b44de0e11281c723ec426f8bec8ca0897ecfe3ff Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm 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