From 7ea1d715f67e81c9fa841c5a62f1cc700ff7363d Mon Sep 17 00:00:00 2001 From: Deepak Mallubhotla Date: Sat, 13 Aug 2022 11:10:00 -0500 Subject: [PATCH] feat: Prints model names while running --- deepdog/real_spectrum_run.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/deepdog/real_spectrum_run.py b/deepdog/real_spectrum_run.py index 03bf147..be4d71a 100644 --- a/deepdog/real_spectrum_run.py +++ b/deepdog/real_spectrum_run.py @@ -140,8 +140,10 @@ class RealSpectrumRun: results = [] _logger.debug("Going to iterate over models now") - for model_count, model in enumerate(self.models): - _logger.debug(f"Doing model #{model_count}") + for model_count, (model, model_name) in enumerate( + zip(self.models, self.model_names) + ): + _logger.debug(f"Doing model #{model_count}: {model_name}") core_count = multiprocessing.cpu_count() - 1 or 1 with multiprocessing.Pool(core_count) as pool: cycle_count = 0