Update imports for trainer

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

TTS/bin/train_tts.py

@@ -1,3 +1,4 @@
+from dataclasses import dataclass, field
 import os
 from dataclasses import dataclass, field
 

TTS/bin/train_vocoder.py

@@ -1,3 +1,4 @@
+from dataclasses import dataclass, field
 import os
 from dataclasses import dataclass, field
 

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