From ef7e31c498f4fdd12acc1aad5ce12fb36d80a970 Mon Sep 17 00:00:00 2001 From: erogol Date: Mon, 25 May 2020 10:32:14 +0200 Subject: [PATCH] fix #412 --- speaker_encoder/train.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/speaker_encoder/train.py b/speaker_encoder/train.py index 0a137360..f74d0880 100644 --- a/speaker_encoder/train.py +++ b/speaker_encoder/train.py @@ -13,12 +13,11 @@ from TTS.speaker_encoder.model import SpeakerEncoder from TTS.speaker_encoder.visual import plot_embeddings from TTS.speaker_encoder.generic_utils import save_best_model from TTS.utils.audio import AudioProcessor -from TTS.utils.generic_utils import (NoamLR, check_update, copy_config_file, - count_parameters, - create_experiment_folder, get_git_branch, - load_config, +from TTS.utils.generic_utils import (create_experiment_folder, get_git_branch, remove_experiment_folder, set_init_dict) -from TTS.utils.logger import Logger +from TTS.utils.io import load_config, copy_config_file +from TTS.utils.training import check_update, NoamLR +from TTS.utils.tensorboard_logger import TensorboardLogger from TTS.utils.radam import RAdam torch.backends.cudnn.enabled = True @@ -237,7 +236,7 @@ if __name__ == '__main__': new_fields) LOG_DIR = OUT_PATH - tb_logger = Logger(LOG_DIR) + tb_logger = TensorboardLogger(LOG_DIR) try: main(args)