Fix (GPFQ): change default value for p to 1

Xilinx · Nov 15, 2023 · b95bbd5 · b95bbd5
1 parent 0c16d66
commit b95bbd5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/brevitas/graph/gpfq.py b/src/brevitas/graph/gpfq.py
@@ -43,7 +43,7 @@ def __init__(
             inplace: bool = True,
             create_weight_orig: bool = True,
             use_quant_activations: bool = True,
-            p: int = 0.25,
+            p: float = 1.0,
             return_forward_output: bool = False,
             act_order: bool = False) -> None:
         if not inplace:
@@ -117,10 +117,10 @@ def __init__(
             act_order,
             len_parallel_layers=1,
             create_weight_orig=True,
-            p=0.25) -> None:
-        
+            p=1.0) -> None:
+
         super().__init__(layer, name, act_order, len_parallel_layers, create_weight_orig)
-        
+
         self.float_input = None
         self.quantized_input = None
         self.index_computed = False

diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py
@@ -167,7 +167,7 @@
     default=True,
     help='Narrow range for weight quantization (default: enabled)')
 parser.add_argument(
-    '--gpfq-p', default=0.25, type=float, help='P parameter for GPFQ (default: 0.25)')
+    '--gpfq-p', default=1.0, type=float, help='P parameter for GPFQ (default: 0.25)')
 parser.add_argument(
     '--quant-format',
     default='int',