From af63b5956da28234a48d6ac8b676d68eeeac2c41 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Thu, 7 Dec 2023 09:56:35 +0000 Subject: [PATCH] Fix (ptq/evaluate): add missing arg type --- .../imagenet_classification/ptq/ptq_evaluate.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py index 6f4243741..01148911e 100644 --- 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',