The definition of 'quantizer' in quantize.py

facebookresearch / kill-the-bits

Code for: "And the bit goes down: Revisiting the quantization of neural networks"

Other

636 stars 123 forks source link

The definition of 'quantizer' in quantize.py #17

Closed yw155 closed 5 years ago

yw155 commented 5 years ago

Hi @pierrestock, I notice that the variable 'quantizer' in line 363 of 'quantize.py' is not defined. Is it correct using the definition in the previous 'for' loop? Thank you.

pierrestock commented 5 years ago

Hi yw155,

Yes, the quantizer we use is just a placeholder in which to load the proper assignments. Then, we compute the finetuned centroids and save both tensors. I agree that the coding of this part may be a bit confusing, sorry for that!

Pierre