From bf540f43237b0b85e0b03d86c40347de42b5c76f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Sun, 20 Feb 2022 11:35:42 +0100 Subject: [PATCH] Update imports for trainer --- TTS/bin/train_encoder.py | 2 ++ TTS/bin/train_tts.py | 1 + TTS/bin/train_vocoder.py | 1 + TTS/speaker_encoder/utils/training.py | 6 +++++- 4 files changed, 9 insertions(+), 1 deletion(-) diff --git a/TTS/bin/train_encoder.py b/TTS/bin/train_encoder.py index 5828411c..33724919 100644 --- a/TTS/bin/train_encoder.py +++ b/TTS/bin/train_encoder.py @@ -10,6 +10,8 @@ import torch from torch.utils.data import DataLoader from trainer.torch import NoamLR +from trainer.torch import NoamLR + from TTS.speaker_encoder.dataset import SpeakerEncoderDataset from TTS.speaker_encoder.losses import AngleProtoLoss, GE2ELoss, SoftmaxAngleProtoLoss from TTS.speaker_encoder.utils.generic_utils import save_best_model, setup_speaker_encoder_model diff --git a/TTS/bin/train_tts.py b/TTS/bin/train_tts.py index 31813712..1bca7430 100644 --- a/TTS/bin/train_tts.py +++ b/TTS/bin/train_tts.py @@ -1,3 +1,4 @@ +from dataclasses import dataclass, field import os from dataclasses import dataclass, field diff --git a/TTS/bin/train_vocoder.py b/TTS/bin/train_vocoder.py index 32ecd7bd..1745d6ab 100644 --- a/TTS/bin/train_vocoder.py +++ b/TTS/bin/train_vocoder.py @@ -1,3 +1,4 @@ +from dataclasses import dataclass, field import os from dataclasses import dataclass, field diff --git a/TTS/speaker_encoder/utils/training.py b/TTS/speaker_encoder/utils/training.py index 0bc72af8..c07915c9 100644 --- a/TTS/speaker_encoder/utils/training.py +++ b/TTS/speaker_encoder/utils/training.py @@ -1,3 +1,5 @@ +from asyncio.log import logger +from dataclasses import dataclass, field import os from dataclasses import dataclass, field @@ -7,10 +9,12 @@ from trainer.logging import logger_factory from trainer.logging.console_logger import ConsoleLogger from TTS.config import load_config, register_config +from trainer import TrainerArgs, get_last_checkpoint from TTS.tts.utils.text.characters import parse_symbols from TTS.utils.generic_utils import get_experiment_folder_path, get_git_branch from TTS.utils.io import copy_model_files -from trainer import get_last_checkpoint +from trainer.logging import logger_factory +from trainer.logging.console_logger import ConsoleLogger @dataclass