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

Fix trainer malformatted print

Browse Source
This commit is contained in:
Eren Gölge
2021-09-13 08:32:02 +00:00
parent 91bebebe18
commit a97dc8d09f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/trainer.py
+1 -1
View File
@@ -880,7 +880,7 @@ class Trainer:
"""Restore the best loss from the args.best_path if provided else
from the model (`args.restore_path` or `args.continue_path`) used for resuming the training"""
if self.restore_step != 0 or self.args.best_path:
print(" > Restoring best loss from " f"{os.path.basename(self.args.best_path)} ...")
print(f" > Restoring best loss from {os.path.basename(self.args.best_path)} ...")
ch = load_fsspec(self.args.restore_path, map_location="cpu")
if "model_loss" in ch:
self.best_loss = ch["model_loss"]