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

Fix logging current learning rate in trainer

This commit is contained in:
Eren Gölge 2021-09-09 09:01:04 +00:00
parent 6c4c1065b0
commit abf5e48177
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
TTS/trainer.py
View File

@ -663,6 +663,7 @@ class Trainer:
lrs = {"current_lr": current_lr} lrs = {"current_lr": current_lr}
# log run-time stats # log run-time stats
loss_dict.update(lrs)
loss_dict.update( loss_dict.update(
{ {
"step_time": round(step_time, 4), "step_time": round(step_time, 4),
@ -1126,7 +1127,7 @@ def get_last_checkpoint(path: str) -> Tuple[str, str]:
last_model_num = model_num last_model_num = model_num
last_model = file_name last_model = file_name
# if there is not checkpoint found above # if there is no checkpoint found above
# find the checkpoint with the latest # find the checkpoint with the latest
# modification date. # modification date.
key_file_names = [fn for fn in file_names if key in fn] key_file_names = [fn for fn in file_names if key in fn]