Fix (examples/ptq): fix for float act quant MSE

Xilinx · May 6, 2024 · a422d0e · a422d0e
1 parent 20510a9
commit a422d0e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
@@ -120,7 +120,7 @@
                     'sym': Fp8e4m3ActPerTensorFloat}},
             'mse': {
                 'per_tensor': {
-                    'sym': Fp8e4m3ActPerTensorFloat},}}}}
+                    'sym': Fp8e4m3ActPerTensorFloatMSE}}}}}
 
 
 def quantize_model(