From b87a7ac356a5a0bf14a16631be5cc047de13a7be Mon Sep 17 00:00:00 2001 From: thllwg Date: Mon, 13 Jul 2020 19:02:17 +0200 Subject: [PATCH] Round seconds to two decimals --- utils/console_logger.py | 2 +- vocoder/utils/console_logger.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/console_logger.py b/utils/console_logger.py index 39da8b08..85d5b376 100644 --- a/utils/console_logger.py +++ b/utils/console_logger.py @@ -56,7 +56,7 @@ class ConsoleLogger(): def print_train_epoch_end(self, global_step, epoch, epoch_time, print_dict): indent = " | > " - log_text = f"\n{tcolors.BOLD} --> TRAIN PERFORMACE -- EPOCH TIME: {epoch_time} sec -- GLOBAL_STEP: {global_step}{tcolors.ENDC}\n" + log_text = f"\n{tcolors.BOLD} --> TRAIN PERFORMACE -- EPOCH TIME: {epoch_time:.2f} sec -- GLOBAL_STEP: {global_step}{tcolors.ENDC}\n" for key, value in print_dict.items(): log_text += "{}{}: {:.5f}\n".format(indent, key, value) print(log_text, flush=True) diff --git a/vocoder/utils/console_logger.py b/vocoder/utils/console_logger.py index 71181956..6af0b823 100644 --- a/vocoder/utils/console_logger.py +++ b/vocoder/utils/console_logger.py @@ -55,7 +55,7 @@ class ConsoleLogger(): def print_train_epoch_end(self, global_step, epoch, epoch_time, print_dict): indent = " | > " - log_text = f"\n{tcolors.BOLD} --> TRAIN PERFORMACE -- EPOCH TIME: {epoch_time} sec -- GLOBAL_STEP: {global_step}{tcolors.ENDC}\n" + log_text = f"\n{tcolors.BOLD} --> TRAIN PERFORMACE -- EPOCH TIME: {epoch_time:.2f} sec -- GLOBAL_STEP: {global_step}{tcolors.ENDC}\n" for key, value in print_dict.items(): log_text += "{}{}: {:.5f}\n".format(indent, key, value) print(log_text, flush=True)