diff --git a/TTS/utils/logging/console_logger.py b/TTS/utils/logging/console_logger.py index bb6644c9..c5fbe8b4 100644 --- a/TTS/utils/logging/console_logger.py +++ b/TTS/utils/logging/console_logger.py @@ -38,7 +38,7 @@ class ConsoleLogger: def print_train_start(self): print(f"\n{tcolors.BOLD} > TRAINING ({self.get_time()}) {tcolors.ENDC}") - def print_train_step(self, batch_steps, step, global_step, log_dict, loss_dict, avg_loss_dict): + def print_train_step(self, batch_steps, step, global_step, loss_dict, avg_loss_dict): indent = " | > " print() log_text = "{} --> STEP: {}/{} -- GLOBAL_STEP: {}{}\n".format( @@ -50,13 +50,6 @@ class ConsoleLogger: log_text += "{}{}: {:.5f} ({:.5f})\n".format(indent, key, value, avg_loss_dict[f"avg_{key}"]) else: log_text += "{}{}: {:.5f} \n".format(indent, key, value) - for idx, (key, value) in enumerate(log_dict.items()): - if isinstance(value, list): - log_text += f"{indent}{key}: {value[0]:.{value[1]}f}" - else: - log_text += f"{indent}{key}: {value}" - if idx < len(log_dict) - 1: - log_text += "\n" print(log_text, flush=True) # pylint: disable=unused-argument