Fixing default weight_quant

Xilinx · Jul 12, 2023 · 984c8ab · 984c8ab
1 parent 2cdf3d4
commit 984c8ab
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/brevitas_examples/bnn_pynq/models/resnet.py b/src/brevitas_examples/bnn_pynq/models/resnet.py
@@ -8,6 +8,7 @@
 
 import brevitas.nn as qnn
 from brevitas.quant import Int8WeightPerChannelFloat
+from brevitas.quant import Int8WeightPerTensorFloat
 from brevitas.quant import TruncTo8bit
 from brevitas.quant_tensor import QuantTensor
 
@@ -121,7 +122,7 @@ def __init__(
             round_average_pool=False,
             weight_quant=Int8WeightPerChannelFloat,
             first_layer_weight_quant=Int8WeightPerChannelFloat,
-            last_layer_weight_quant=Int8WeightPerChannelFloat):
+            last_layer_weight_quant=Int8WeightPerTensorFloat):
         super(QuantResNet, self).__init__()
         self.in_planes = 64
         self.conv1 = make_quant_conv2d(