Fix (quant/minifloat): add scaling_min_val to base quantizers (#773)

Xilinx · Dec 5, 2023 · 422b632 · 422b632
1 parent 58936c4
commit 422b632
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/brevitas/quant/experimental/float_base.py b/src/brevitas/quant/experimental/float_base.py
@@ -17,13 +17,15 @@ class FloatWeightBase(SolveTensorQuantFloatToIntImplFromEnum):
     tensor_quant = FloatQuant
     signed = True
     float_to_int_impl_type = 'round'
+    scaling_min_val = 1e-10
 
 
 class FloatActBase(SolveTensorQuantFloatToIntImplFromEnum):
     proxy_class = ActQuantProxyFromInjector
     tensor_quant = FloatQuant
     signed = True
     float_to_int_impl_type = 'round'
+    scaling_min_val = 1e-10
 
 
 class ScaledFloatWeightBase(FloatWeightBase, WeightQuantSolver):