From 9bbc9243770225634b18a1e2a0372a3f95e354ba Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Fri, 28 May 2021 13:32:35 +0200 Subject: [PATCH] import missings --- TTS/trainer.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) 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