Add config_file property when initializing QuantSim object

Signed-off-by: Geunho Lee <[email protected]>
quic · Jan 26, 2024 · 2d794ef · 2d794ef
1 parent 5fe5d2d
commit 2d794ef
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/TrainingExtensions/torch/src/python/aimet_torch/quantsim.py b/TrainingExtensions/torch/src/python/aimet_torch/quantsim.py
@@ -204,6 +204,7 @@ def __init__(self, model: torch.nn.Module, dummy_input: Union[torch.Tensor, Tupl
         self._rounding_mode = rounding_mode
         self._default_output_bw = default_output_bw
         self._default_param_bw = default_param_bw
+        self._config_file = config_file
         self._is_conditional = False
         self._module_marker_map = {}
         self._percentile_value = 100 # default percentile value