Skip to content

Commit e41d1b4

Browse files
authored
Information in logs if Manual or Automatic config search is used (#759)
* Log brute config search mode * Fix duplicate logging
1 parent 228062b commit e41d1b4

File tree

3 files changed

+16
-3
lines changed

3 files changed

+16
-3
lines changed

model_analyzer/config/generate/automatic_model_config_generator.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@
3636
class AutomaticModelConfigGenerator(BaseModelConfigGenerator):
3737
"""Given a model, generates model configs in automatic search mode"""
3838

39+
_log_first_run = False
40+
3941
def __init__(
4042
self,
4143
config: ConfigCommandProfile,
@@ -71,6 +73,12 @@ def __init__(
7173
early_exit_enable,
7274
)
7375

76+
if not AutomaticModelConfigGenerator._log_first_run:
77+
logger.info("")
78+
logger.info("Starting automatic brute search")
79+
logger.info("")
80+
AutomaticModelConfigGenerator._log_first_run = True
81+
7482
self._max_instance_count = config.run_config_search_max_instance_count
7583
self._min_instance_count = config.run_config_search_min_instance_count
7684
self._max_model_batch_size = config.run_config_search_max_model_batch_size

model_analyzer/config/generate/brute_plus_binary_parameter_search_run_config_generator.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -89,9 +89,6 @@ def get_configs(self) -> Generator[RunConfig, None, None]:
8989
The next RunConfig generated by this class
9090
"""
9191

92-
logger.info("")
93-
logger.info("Starting brute mode search")
94-
logger.info("")
9592
yield from self._execute_brute_search()
9693
logger.info("")
9794
logger.info("Done with brute mode search.")

model_analyzer/config/generate/manual_model_config_generator.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,8 @@
3838
class ManualModelConfigGenerator(BaseModelConfigGenerator):
3939
"""Given a model, generates model configs in manual search mode"""
4040

41+
_log_first_run = False
42+
4143
def __init__(
4244
self,
4345
config: ConfigCommandProfile,
@@ -72,6 +74,12 @@ def __init__(
7274
early_exit_enable,
7375
)
7476

77+
if not ManualModelConfigGenerator._log_first_run:
78+
logger.info("")
79+
logger.info("Starting manual brute search")
80+
logger.info("")
81+
ManualModelConfigGenerator._log_first_run = True
82+
7583
self._search_disabled = config.run_config_search_disable
7684
self._curr_config_index = 0
7785
self._curr_max_batch_size_index = 0

0 commit comments

Comments
 (0)