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

more console printing formatting

This commit is contained in:
erogol 2020-04-29 11:58:51 +02:00
parent 091711459d
commit 67420eeb86
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -586,7 +586,7 @@ def main(args): # pylint: disable=redefined-outer-name
model.decoder.set_r(r) model.decoder.set_r(r)
if c.bidirectional_decoder: if c.bidirectional_decoder:
model.decoder_backward.set_r(r) model.decoder_backward.set_r(r)
print("\n > Number of output frames:", model.decoder.r) print("\n > Number of output frames:", model.decoder.r)
train_avg_loss_dict, global_step = train(model, criterion, optimizer, train_avg_loss_dict, global_step = train(model, criterion, optimizer,
optimizer_st, scheduler, ap, optimizer_st, scheduler, ap,

5
utils/console_logger.py
View File

@ -62,10 +62,11 @@ class ConsoleLogger():
print(log_text, flush=True) print(log_text, flush=True)
def print_eval_start(self): def print_eval_start(self):
print(f"{tcolors.BOLD} > EVALUATION {tcolors.ENDC}\n") print(f"{tcolors.BOLD} > EVALUATION {tcolors.ENDC}")
def print_eval_step(self, step, loss_dict, avg_loss_dict): def print_eval_step(self, step, loss_dict, avg_loss_dict):
indent = " | > " indent = " | > "
print()
log_text = f"{tcolors.BOLD} --> STEP: {step}{tcolors.ENDC}\n" log_text = f"{tcolors.BOLD} --> STEP: {step}{tcolors.ENDC}\n"
for key, value in loss_dict.items(): for key, value in loss_dict.items():
# print the avg value if given # print the avg value if given
@ -77,7 +78,7 @@ class ConsoleLogger():
def print_epoch_end(self, epoch, avg_loss_dict): def print_epoch_end(self, epoch, avg_loss_dict):
indent = " | > " indent = " | > "
log_text = " {}--> EVAL PERFORMANCE{}\n".format( log_text = "\n {}--> EVAL PERFORMANCE{}\n".format(
tcolors.BOLD, tcolors.ENDC) tcolors.BOLD, tcolors.ENDC)
for key, value in avg_loss_dict.items(): for key, value in avg_loss_dict.items():
# print the avg value if given # print the avg value if given