search

lucidrains / vector-quantize-pytorch

Vector (and Scalar) Quantization, in Pytorch
MIT License
2.66k stars 216 forks source link

[BUG] ResidualVQ always initializes MLPs #160

Closed hummat closed 2 months ago

hummat commented 2 months ago

MLPs in ResidualVQ are always initialized, even if implicit_neural_codebook=False.

https://github.com/lucidrains/vector-quantize-pytorch/blob/54d29e8fba72443a29928e9f94ee52ca43eeb60a/vector_quantize_pytorch/residual_vq.py#L153

lucidrains commented 2 months ago

@hummat oops, yes indeed

thank you Matthias!