From c6b86cc4ae6b1498e2f6cbbbdf5ef2d7ac9db929 Mon Sep 17 00:00:00 2001 From: Alessandro Pappalardo Date: Thu, 21 Sep 2023 17:24:10 +0100 Subject: [PATCH] Feat (quant): switch custom float scaling default to absmax --- src/brevitas/quant/experimental/float_base.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/brevitas/quant/experimental/float_base.py b/src/brevitas/quant/experimental/float_base.py index f1d191057..cb2d3ae22 100644 --- a/src/brevitas/quant/experimental/float_base.py +++ b/src/brevitas/quant/experimental/float_base.py @@ -33,10 +33,9 @@ class ScaledFloatWeightBase(FloatWeightBase, WeightQuantSolver): class ScaledFloatActBase(FloatActBase, ActQuantSolver): - scaling_stats_op = 'percentile' + scaling_stats_op = 'max' scaling_impl_type = 'parameter_from_stats' restrict_scaling_type = 'fp' - high_percentile_q = 99.999 collect_stats_steps = 300 float_scaling_impl = FloatScaling