Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplify `console_logger`

This commit is contained in:
Eren Gölge 2021-08-07 21:54:19 +00:00
parent dd55960732
commit e94c1f894d
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/utils/logging/console_logger.py
View File

@ -38,7 +38,7 @@ class ConsoleLogger:
def print_train_start(self): def print_train_start(self):
print(f"\n{tcolors.BOLD} > TRAINING ({self.get_time()}) {tcolors.ENDC}") 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 = " | > " indent = " | > "
print() print()
log_text = "{} --> STEP: {}/{} -- GLOBAL_STEP: {}{}\n".format( 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}"]) log_text += "{}{}: {:.5f} ({:.5f})\n".format(indent, key, value, avg_loss_dict[f"avg_{key}"])
else: else:
log_text += "{}{}: {:.5f} \n".format(indent, key, value) 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) print(log_text, flush=True)
# pylint: disable=unused-argument # pylint: disable=unused-argument