Update imports

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

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
 

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

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