Exemplo n.º 1
0
 def prepare_bias(cls, module: QuantWBIOL):
     if module.bias is not None and not module.is_bias_quant_enabled:
         bias = module.bias.detach()
     elif module.bias is not None and module.is_bias_quant_enabled:
         bias = module.quant_bias()
     else:
         bias = module.bias
     return bias
Exemplo n.º 2
0
 def prepare_bias(cls, module: QuantWBIOL):
     if module.bias is not None and not module.is_bias_quant_enabled:
         bias = module.bias.detach()
     elif module.bias is not None and module.is_bias_quant_enabled:
         # export the dequantized value
         bias = module.quant_bias().value
     else:
         bias = module.bias
     return bias