Function get_named_linears

quantization/pre_quant.py:16–17

(module)

def get_named_linears(module):
    return {name: m for name, m in module.named_modules() if isinstance(m, nn.Linear)}

def set_op_by_name(layer, name, new_module):
    levels = name.split('.')