diff --git a/config.json b/config.json index 0b20cd3a..23868a33 100644 --- a/config.json +++ b/config.json @@ -29,7 +29,7 @@ "num_mels": 80, // size of the mel spec frame. "mel_fmin": 0.0, // minimum freq level for mel-spec. ~50 for male and ~95 for female voices. Tune for dataset!! "mel_fmax": 8000.0, // maximum freq level for mel-spec. Tune for dataset!! - "spec_gain": 20, + "spec_gain": 20.0, // Normalization parameters "signal_norm": true, // normalize spec values. Mean-Var normalization if 'stats_path' is defined otherwise range normalization defined by the other params. @@ -106,7 +106,7 @@ // TENSORBOARD and LOGGING "print_step": 25, // Number of steps to log training on console. - "tb_plot_step:": 100, // Number of steps to plot TB training figures. + "tb_plot_step": 100, // Number of steps to plot TB training figures. "print_eval": false, // If True, it prints intermediate loss values in evalulation. "save_step": 10000, // Number of training steps expected to save traninpg stats and checkpoints. "checkpoint": true, // If true, it saves checkpoints per "save_step" diff --git a/utils/console_logger.py b/utils/console_logger.py index 0b361bb8..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} 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 4f44986c..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} 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)