From 83c5ddc5b70fa8c2ddd79a85c4c4a48987dd5502 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Tue, 22 Feb 2022 11:30:41 +0100 Subject: [PATCH] Update imports --- TTS/speaker_encoder/utils/training.py | 2 +- TTS/tts/models/vits.py | 2 +- TTS/vocoder/models/gan.py | 2 +- TTS/vocoder/models/wavegrad.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/TTS/speaker_encoder/utils/training.py b/TTS/speaker_encoder/utils/training.py index c64c46b7..0bc72af8 100644 --- a/TTS/speaker_encoder/utils/training.py +++ b/TTS/speaker_encoder/utils/training.py @@ -10,7 +10,7 @@ from TTS.config import load_config, register_config 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 TTS.utils.trainer_utils import get_last_checkpoint +from trainer import get_last_checkpoint @dataclass diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index 02542f71..0c795ca1 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -29,7 +29,7 @@ from TTS.tts.utils.synthesis import synthesis from TTS.tts.utils.text.characters import BaseCharacters, _characters, _pad, _phonemes, _punctuations from TTS.tts.utils.text.tokenizer import TTSTokenizer from TTS.tts.utils.visual import plot_alignment -from TTS.utils.trainer_utils import get_optimizer, get_scheduler +from trainer.trainer_utils import get_optimizer, get_scheduler from TTS.vocoder.models.hifigan_generator import HifiganGenerator from TTS.vocoder.utils.generic_utils import plot_results diff --git a/TTS/vocoder/models/gan.py b/TTS/vocoder/models/gan.py index d4abaa0a..91467956 100644 --- a/TTS/vocoder/models/gan.py +++ b/TTS/vocoder/models/gan.py @@ -10,7 +10,7 @@ from torch.utils.data.distributed import DistributedSampler from TTS.utils.audio import AudioProcessor from TTS.utils.io import load_fsspec -from TTS.utils.trainer_utils import get_optimizer, get_scheduler +from trainer.trainer_utils import get_optimizer, get_scheduler from TTS.vocoder.datasets.gan_dataset import GANDataset from TTS.vocoder.layers.losses import DiscriminatorLoss, GeneratorLoss from TTS.vocoder.models import setup_discriminator, setup_generator diff --git a/TTS/vocoder/models/wavegrad.py b/TTS/vocoder/models/wavegrad.py index 750258af..95aa3cd2 100644 --- a/TTS/vocoder/models/wavegrad.py +++ b/TTS/vocoder/models/wavegrad.py @@ -10,7 +10,7 @@ from torch.utils.data import DataLoader from torch.utils.data.distributed import DistributedSampler from TTS.utils.io import load_fsspec -from TTS.utils.trainer_utils import get_optimizer, get_scheduler +from trainer.trainer_utils import get_optimizer, get_scheduler from TTS.vocoder.datasets import WaveGradDataset from TTS.vocoder.layers.wavegrad import Conv1d, DBlock, FiLM, UBlock from TTS.vocoder.models.base_vocoder import BaseVocoder