ste_round() usage in quant_modules

Zhen-Dong / HAWQ

Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.

MIT License

410 stars 83 forks source link

ste_round() usage in quant_modules #44

Open balditommaso opened 3 months ago

balditommaso commented 3 months ago

I wonder why the method ste_round() is only used in the QuantLinear Module and not in the others. I would like to ask also why the forward method of the QuantLinear method is not returning also the weight scale factor.