Feat (core): set default MSE iters to 20

Xilinx · Sep 28, 2023 · 4b9970a · 4b9970a
1 parent 2f1709d
commit 4b9970a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/brevitas/core/stats/stats_op.py b/src/brevitas/core/stats/stats_op.py
@@ -436,8 +436,8 @@ def __init__(
             mse_init_op,
             inner_stats_input_view_shape_impl: torch.nn.Module,
             stats_reduce_dim: Optional[int] = None,
-            mse_search_method='fibonacci',
-            mse_iters=10):
+            mse_search_method: str = 'fibonacci',
+            mse_iters: int = 20):
         super(MSE, self).__init__()
         self.mse_init_op = mse_init_op
         self.input_view_shape_impl = inner_stats_input_view_shape_impl