Fix: (examples/resnet18): restore default weight_quant for QuantLinea…

…r to per tensor (#660)
Xilinx · Jul 14, 2023 · 531543e · 531543e
1 parent 2626a79
commit 531543e
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(