Update calibrate.py

Xilinx · Jan 6, 2025 · 471d09f · 471d09f
1 parent b45f298
commit 471d09f
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/src/brevitas/graph/calibrate.py b/src/brevitas/graph/calibrate.py
@@ -75,11 +75,6 @@ def set_collect_stats_to_average(module):
 
 
 def finalize_collect_stats(module):
-    if hasattr(module, 'collect_stats_steps') and hasattr(module, 'counter'):
-        # If the counter has already reached collect_stats_steps, we do not want to reset it
-        # otherwise the restrict_preprocess might be applied twice: during calibration
-        # (that happens in training mode) and then when the model is evaluated
-        module.counter = max(module.collect_stats_steps, module.counter)
     if hasattr(module, 'init_scale'):
         module.init_scale()