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 trainer malformatted print

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/trainer.py
View File

@ -880,7 +880,7 @@ class Trainer:
"""Restore the best loss from the args.best_path if provided else """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""" 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: 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") ch = load_fsspec(self.args.restore_path, map_location="cpu")
if "model_loss" in ch: if "model_loss" in ch:
self.best_loss = ch["model_loss"] self.best_loss = ch["model_loss"]