diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py index 807f32bb828d9a37b8fa94c9e6bdedd7dfeabf90..6f78a03d67a0698274eb4795bc8941c590386063 100644 --- a/mlair/run_modules/post_processing.py +++ b/mlair/run_modules/post_processing.py @@ -390,7 +390,7 @@ class PostProcessing(RunEnvironment): use_multiprocessing=True, verbose=0, steps=1) path = self.data_store.get("model_path") with open(os.path.join(path, "test_scores.txt"), "a") as f: - for index, item in enumerate(test_score): + for index, item in enumerate(to_list(test_score)): logging.info(f"{self.model.metrics_names[index]} (test), {item}") f.write(f"{self.model.metrics_names[index]}, {item}\n") diff --git a/mlair/run_modules/training.py b/mlair/run_modules/training.py index bbb3fabf4dd7c62a71db04cc982544b61d097240..d4badfe25c94133a53a93ca69f1b2f63a955803c 100644 --- a/mlair/run_modules/training.py +++ b/mlair/run_modules/training.py @@ -16,6 +16,7 @@ from mlair.model_modules.keras_extensions import CallbackHandler from mlair.plotting.training_monitoring import PlotModelHistory, PlotModelLearningRate from mlair.run_modules.run_environment import RunEnvironment from mlair.configuration import path_config +from mlair.helpers import to_list class Training(RunEnvironment): @@ -249,5 +250,5 @@ class Training(RunEnvironment): tablefmt="github") val_score = self.model.evaluate_generator(generator=self.val_set, use_multiprocessing=True, verbose=0, steps=1) - for index, item in enumerate(val_score): + for index, item in enumerate(to_list(val_score)): logging.info(f"{self.model.metrics_names[index]} (val), {item}")