Use same codebook size in VQ-VAE example and generative model example

uhh-pd-ml · May 28, 2024 · 51edf04 · 51edf04
1 parent 78748d0
commit 51edf04
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/configs/experiment/example_experiment_tokenization.yaml b/configs/experiment/example_experiment_tokenization.yaml
@@ -76,7 +76,7 @@ model:
     num_heads: 8
     alpha: 10
     vq_kwargs:
-      num_codes: 512
+      num_codes: 8192
       beta: 0.9
       kmeans_init: true
       norm: null