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

Fix logging current learning rate in trainer

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
TTS/trainer.py
+2 -1
View File
@@ -663,6 +663,7 @@ class Trainer:
lrs = {"current_lr": current_lr}
# log run-time stats
loss_dict.update(lrs)
loss_dict.update(
{
"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 = file_name
# if there is not checkpoint found above
# if there is no checkpoint found above
# find the checkpoint with the latest
# modification date.
key_file_names = [fn for fn in file_names if key in fn]