From 3dc7dfa6ad6e6afc1b49a5578c625e20e3401b6f Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Fri, 13 Sep 2024 13:13:59 +0100 Subject: [PATCH] precommit fix --- .../imagenet_classification/ptq/ptq_common.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py index a6cdd2af7..9e5c90e26 100644 --- a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py +++ b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py @@ -178,7 +178,8 @@ class CNNInt8DynamicActPerTensorFloat(Int8DynamicActPerTensorFloat): 'asym': CNNShiftedUint8DynamicActPerTensorFloat}}}, 'po2_scale': { 'stats': { - 'per_group': {'sym':MXInt8Act} }}}}, + 'per_group': { + 'sym': MXInt8Act}}}}}, 'float': { 'static': { 'float_scale': {