Function set_quant_state

quantize/utils.py:69–72

(model, weight_quant: bool = False)

    return total

def set_quant_state(model, weight_quant: bool = False):
    for m in model.modules():
        if isinstance(m, QuantLinear):
            m.set_quant_state(weight_quant)
            
@torch.no_grad()   
def quant_inplace(model):