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