wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

update console logger

Browse Source
This commit is contained in:
Eren Gölge
2021-06-28 17:03:19 +02:00
parent f4f83b6379
commit 178eccbc16
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/utils/console_logger.py
+2 -3
View File
@@ -68,11 +68,10 @@ class ConsoleLogger:
print(log_text, flush=True)
def print_eval_start(self):
print(f"{tcolors.BOLD} > EVALUATION {tcolors.ENDC}\n")
print(f"\n{tcolors.BOLD} > EVALUATION {tcolors.ENDC}\n")
def print_eval_step(self, step, loss_dict, avg_loss_dict):
indent = " | > "
print()
log_text = f"{tcolors.BOLD} --> STEP: {step}{tcolors.ENDC}\n"
for key, value in loss_dict.items():
# print the avg value if given
@@ -84,7 +83,7 @@ class ConsoleLogger:
def print_epoch_end(self, epoch, avg_loss_dict):
indent = " | > "
log_text = " {}--> EVAL PERFORMANCE{}\n".format(tcolors.BOLD, tcolors.ENDC)
log_text = "\n {}--> EVAL PERFORMANCE{}\n".format(tcolors.BOLD, tcolors.ENDC)
for key, value in avg_loss_dict.items():
# print the avg value if given
color = ""