Function quant_inplace

quantize/utils.py:75–78

(model)

            
@torch.no_grad()   
def quant_inplace(model):
    for name, module in model.named_modules():
        if isinstance(module, QuantLinear):
            module.weight.data = module.weight_quantizer(module.weight.data)


class TruncateFunction(torch.autograd.Function):