mirror of https://github.com/coqui-ai/TTS.git
console logger fix
This commit is contained in:
parent
f75b0a6439
commit
97cd39bf99
|
@ -86,10 +86,10 @@ class ConsoleLogger():
|
||||||
sign = '+'
|
sign = '+'
|
||||||
diff = 0
|
diff = 0
|
||||||
if self.old_eval_loss_dict is not None:
|
if self.old_eval_loss_dict is not None:
|
||||||
diff = self.old_eval_loss_dict[key] - value
|
diff = value - self.old_eval_loss_dict[key]
|
||||||
if diff > 0:
|
if diff < 0:
|
||||||
color = tcolors.OKGREEN
|
color = tcolors.OKGREEN
|
||||||
sign = '-'
|
sign = ''
|
||||||
log_text += "{}{}:{} {:.5f} {}({}{:.5f})\n".format(indent, key, color, value, tcolors.ENDC, sign, diff)
|
log_text += "{}{}:{} {:.5f} {}({}{:.5f})\n".format(indent, key, color, value, tcolors.ENDC, sign, diff)
|
||||||
self.old_eval_loss_dict = avg_loss_dict
|
self.old_eval_loss_dict = avg_loss_dict
|
||||||
print(log_text, flush=True)
|
print(log_text, flush=True)
|
Loading…
Reference in New Issue