diff --git a/TTS/trainer.py b/TTS/trainer.py
index 7a31bb34..372bb0f6 100644
--- a/TTS/trainer.py
+++ b/TTS/trainer.py
@@ -7,23 +7,29 @@ import time
 
 import torch
 
+from coqpit import Coqpit
+from dataclasses import dataclass, field
+from typing import Tuple, Dict, List, Union
+
+from argparse import Namespace
 # DISTRIBUTED
+from torch import nn
 from torch.nn.parallel import DistributedDataParallel as DDP_th
 from torch.utils.data import DataLoader
 from torch.utils.data.distributed import DistributedSampler
 
+from TTS.utils.logging import ConsoleLogger, TensorboardLogger
 from TTS.tts.datasets import TTSDataset, load_meta_data
 from TTS.tts.layers import setup_loss
 from TTS.tts.models import setup_model
 from TTS.tts.utils.io import save_best_model, save_checkpoint
 from TTS.tts.utils.speakers import SpeakerManager
 from TTS.tts.utils.synthesis import synthesis
-from TTS.tts.utils.text.symbols import make_symbols, phonemes, symbols
+from TTS.tts.utils.text.symbols import make_symbols
 from TTS.tts.utils.visual import plot_alignment, plot_spectrogram
-from TTS.utils.arguments import init_training
 from TTS.utils.audio import AudioProcessor
-from TTS.utils.distribute import init_distributed, reduce_tensor
-from TTS.utils.generic_utils import KeepAverage, count_parameters, find_module, remove_experiment_folder, set_init_dict
+from TTS.utils.distribute import init_distributed
+from TTS.utils.generic_utils import KeepAverage, count_parameters, set_init_dict
 from TTS.utils.training import check_update, setup_torch_training_env