Fix (ptq/evaluate): add missing arg type

Xilinx · Dec 7, 2023 · af63b59 · af63b59
1 parent 69d3a1e
commit af63b59
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py
@@ -92,7 +92,11 @@
     type=int,
     help='Input and weights bit width for first and last layer w/ layerwise backend (default: 8)')
 parser.add_argument(
-    '--bias-bit-width', default=32, choices=[32, 16, None], help='Bias bit width (default: 32)')
+    '--bias-bit-width',
+    default=32,
+    type=int,
+    choices=[32, 16, None],
+    help='Bias bit width (default: 32)')
 parser.add_argument(
     '--act-quant-type',
     default='sym',