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',