haq
haq copied to clipboard
Linear quantization
trafficstars
Can you please point to the part where the linear quantization is used? Thanks. I can't find it in linear quantize environment. self.quantizable_layer_types = [QConv2d, QLinear] might be [nn.Conv2d, nn.Linear]?
[QConv2d, QLinear] are linear quantization layers. And you can find the implementation in https://github.com/mit-han-lab/haq/blob/7141586e9ae47c8a50aa8b596ab37682a06b434a/lib/utils/quantize_utils.py#L121