diff options
Diffstat (limited to 'training/conf/network/vqvae.yaml')
| -rw-r--r-- | training/conf/network/vqvae.yaml | 18 |
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 |