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 loading always best_model when continue

This commit is contained in:
Eren Gölge 2021-04-01 03:41:15 +02:00
parent f07ca29386
commit e3c052382b
1 changed files with 2 additions and 1 deletions
TTS/utils
arguments.py

3
TTS/utils/arguments.py
View File

@ -7,6 +7,7 @@ import glob
import os import os
import re import re
import torch
from TTS.tts.utils.text.symbols import parse_symbols from TTS.tts.utils.text.symbols import parse_symbols
from TTS.utils.console_logger import ConsoleLogger from TTS.utils.console_logger import ConsoleLogger
from TTS.utils.generic_utils import create_experiment_folder, get_git_branch from TTS.utils.generic_utils import create_experiment_folder, get_git_branch
@ -103,7 +104,7 @@ def get_last_checkpoint(path):
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]
if last_model is None and len(key_file_names) > 0: if last_model is None and len(key_file_names) > 0:
last_model = max(key_file_names, key=os.path.getctime) last_model = max(key_file_names, key=os.path.getctime)
last_model_num = os.path.getctime(last_model) last_model_num = torch.load(last_model)['step']
if last_model is not None: if last_model is not None:
last_models[key] = last_model last_models[key] = last_model