Bugfix (nn): remove leftover FloatBias

Signed-off-by: Alessandro Pappalardo <[email protected]>
nickfraser · Jan 29, 2021 · a61646e · a61646e
1 parent 75c3e1b
commit a61646e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/brevitas/nn/quant_conv.py b/src/brevitas/nn/quant_conv.py
@@ -166,7 +166,7 @@ def __init__(
             bias: bool = True,
             padding_type: str = 'standard',
             weight_quant: Union[WeightQuantProxyProtocol, Type[Injector]] = Int8WeightPerTensorFloat,
-            bias_quant: Union[BiasQuantProxyProtocol, Type[Injector]] = FloatBias,
+            bias_quant: Union[BiasQuantProxyProtocol, Type[Injector]] = None,
             input_quant: Union[ActQuantProxyProtocol, Type[Injector]] = None,
             output_quant: Union[ActQuantProxyProtocol, Type[Injector]] = None,
             return_quant_tensor: bool = False,

diff --git a/src/brevitas/nn/quant_convtranspose.py b/src/brevitas/nn/quant_convtranspose.py
@@ -159,7 +159,7 @@ def __init__(
             groups: int = 1,
             bias: bool = True,
             weight_quant: Union[WeightQuantProxyProtocol, Type[Injector]] = Int8WeightPerTensorFloat,
-            bias_quant: Union[BiasQuantProxyProtocol, Type[Injector]] = FloatBias,
+            bias_quant: Union[BiasQuantProxyProtocol, Type[Injector]] = None,
             input_quant: Union[ActQuantProxyProtocol, Type[Injector]] = None,
             output_quant: Union[ActQuantProxyProtocol, Type[Injector]] = None,
             return_quant_tensor: bool = False,