search

lucidrains / vector-quantize-pytorch

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

Check if we actually training #163

Closed inspirit closed 1 month ago

inspirit commented 1 month ago

Hi Phil,

I believe we need to make sure we are training RVQ before updating codebooks: https://github.com/lucidrains/vector-quantize-pytorch/blob/22a03755f3acc1e011ecf37706496eb249f68f90/vector_quantize_pytorch/residual_vq.py#L370

lucidrains commented 1 month ago

@inspirit ah yes indeed, thanks!