diff --git a/vocoder/utils/console_logger.py b/vocoder/utils/console_logger.py index ff7edfd0..50882160 100644 --- a/vocoder/utils/console_logger.py +++ b/vocoder/utils/console_logger.py @@ -15,8 +15,8 @@ tcolors = AttrDict({ class ConsoleLogger(): + # TODO: merge this with TTS ConsoleLogger def __init__(self): - # TODO: color code for value changes # use these to compare values between iterations self.old_train_loss_dict = None self.old_epoch_loss_dict = None @@ -36,7 +36,7 @@ class ConsoleLogger(): print(f"\n{tcolors.BOLD} > TRAINING ({self.get_time()}) {tcolors.ENDC}") def print_train_step(self, batch_steps, step, global_step, - step_time, loader_time, lr, + step_time, loader_time, lrG, lrD, loss_dict, avg_loss_dict): indent = " | > " print() @@ -48,7 +48,7 @@ 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) - log_text += f"{indent}step_time: {step_time:.2f}\n{indent}loader_time: {loader_time:.2f}\n{indent}lr: {lr:.5f}" + log_text += f"{indent}step_time: {step_time:.2f}\n{indent}loader_time: {loader_time:.2f}\n{indent}lrG: {lrG}\n{indent}lrD: {lrD}" print(log_text, flush=True) # pylint: disable=unused-argument