From 2202e171c53dde2359e95c7fa82bd1c7686604de Mon Sep 17 00:00:00 2001 From: krzim <44442910+krzim@users.noreply.github.com> Date: Thu, 29 Oct 2020 18:03:11 -0400 Subject: [PATCH] Fix import to grab the encoder model save function I saw that this was recently changed but I'm not sure if it should have been. This is the correct function given the arguments provided to it in the train loop. --- TTS/bin/train_encoder.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/TTS/bin/train_encoder.py b/TTS/bin/train_encoder.py index 8d1f14fa..6a370362 100644 --- a/TTS/bin/train_encoder.py +++ b/TTS/bin/train_encoder.py @@ -13,10 +13,9 @@ from TTS.speaker_encoder.dataset import MyDataset from TTS.speaker_encoder.losses import AngleProtoLoss, GE2ELoss from TTS.speaker_encoder.model import SpeakerEncoder from TTS.speaker_encoder.utils.generic_utils import \ - check_config_speaker_encoder + check_config_speaker_encoder, save_best_model from TTS.speaker_encoder.utils.visual import plot_embeddings from TTS.tts.datasets.preprocess import load_meta_data -from TTS.tts.utils.io import save_best_model from TTS.utils.audio import AudioProcessor from TTS.utils.generic_utils import (count_parameters, create_experiment_folder, get_git_branch,