diff --git a/src/brevitas/graph/gpfq.py b/src/brevitas/graph/gpfq.py index 995744fc5..38a9e9993 100644 --- a/src/brevitas/graph/gpfq.py +++ b/src/brevitas/graph/gpfq.py @@ -115,9 +115,10 @@ def initialize_module_optimizer( act_order=act_order, len_parallel_layers=len_parallel_layers, create_weight_orig=create_weight_orig, - p=self.p + p=self.p, accumulator_bit_width=self.accumulator_bit_width) + class GPFQ(GPxQ): """ Based on https://github.com/YixuanSeanZhou/Quantized_Neural_Nets/tree/main