diff --git a/src/brevitas_examples/bnn_pynq/models/resnet.py b/src/brevitas_examples/bnn_pynq/models/resnet.py index 309431d2f..24995c6b6 100644 --- a/src/brevitas_examples/bnn_pynq/models/resnet.py +++ b/src/brevitas_examples/bnn_pynq/models/resnet.py @@ -9,8 +9,8 @@ import brevitas.nn as qnn from brevitas.quant import Int8WeightPerChannelFloat from brevitas.quant import Int8WeightPerTensorFloat -from brevitas.quant import TruncTo8bit from brevitas.quant import IntBias +from brevitas.quant import TruncTo8bit from brevitas.quant_tensor import QuantTensor @@ -232,4 +232,4 @@ def quant_resnet18(cfg) -> QuantResNet: num_classes=num_classes, weight_bit_width=weight_bit_width, act_bit_width=act_bit_width) - return model \ No newline at end of file + return model