Merge pull request #156 from lweitkamp/master

Remove double l2 normalization

Author: lucidrains

vector_quantize_pytorch/vector_quantize_pytorch.py

@@ -751,7 +751,7 @@ def forward(
             embed_normalized = self.embed_avg / rearrange(cluster_size, '... -> ... 1')
             embed_normalized = l2norm(embed_normalized)
 
-            self.embed.data.copy_(l2norm(embed_normalized))
+            self.embed.data.copy_(embed_normalized)
             self.expire_codes_(x)
 
         if needs_codebook_dim: