Enhance code readability

openvinotoolkit · Aug 24, 2022 · e8b32a1 · e8b32a1
1 parent 9531841
commit e8b32a1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/external/model-preparation-algorithm/mpa_tasks/apis/classification/task.py b/external/model-preparation-algorithm/mpa_tasks/apis/classification/task.py
@@ -305,8 +305,9 @@ def _init_model_cfg(self):
         if self._multilabel:
             template = MPAConfig.fromfile(self.template_file_path)
             template_params = template.hyper_parameters.parameter_overrides.learning_parameters
-            if cfg.get('runner', False) and (template_params.num_iters.default_value != self._hyperparams.learning_parameters.num_iters):
-                cfg.runner.max_epochs = self._hyperparams.learning_parameters.num_iters
+            incoming_params = self._hyperparams.learning_parameters
+            if cfg.get('runner', False) and (template_params.num_iters.default_value != incoming_params.num_iters):
+                cfg.runner.max_epochs = incoming_params.num_iters
 
         cfg.model.multilabel = self._multilabel
         cfg.model.hierarchical = self._hierarchical