Merge pull request #150 from dwromero/master

Added access to codebook_size and codebook_dim as properties in ResidualVQ

Author: lucidrains

vector_quantize_pytorch/residual_vq.py

@@ -84,6 +84,14 @@ def __init__(
 
         for vq in rest_vq:
             vq._codebook = codebook
+    
+    @property
+    def codebook_size(self):
+        return self.layers[0].codebook_size
+    
+    @property
+    def codebook_dim(self):
+        return self.layers[0].codebook_dim
 
     @property
     def codebooks(self):