Fix (calibrate): avoid suppressing checkpoint errors

Xilinx · Oct 1, 2024 · 1094eea · 1094eea
1 parent 1e71a04
commit 1094eea
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/brevitas/graph/calibrate.py b/src/brevitas/graph/calibrate.py
@@ -149,7 +149,6 @@ def __exit__(self, *args, **kwargs):
         for module in self.model.modules():
             if issubclass(type(module), QuantWBIOL):
                 module._quant_load_model_mode = False
-        return True
 
 
 class ClipFloatWeights(Transform):