From 0ee0458309dd88352ea3280d7ebb28561d3905b1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Thu, 8 Apr 2021 11:29:15 +0200 Subject: [PATCH] remove redundant imports --- TTS/bin/train_vocoder_gan.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/TTS/bin/train_vocoder_gan.py b/TTS/bin/train_vocoder_gan.py index 0af49c1f..000a545d 100644 --- a/TTS/bin/train_vocoder_gan.py +++ b/TTS/bin/train_vocoder_gan.py @@ -9,13 +9,15 @@ import traceback from inspect import signature import torch +# DISTRIBUTED +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.arguments import parse_arguments, process_args from TTS.utils.audio import AudioProcessor +from TTS.utils.distribute import init_distributed from TTS.utils.generic_utils import (KeepAverage, count_parameters, remove_experiment_folder, set_init_dict) -from TTS.utils.io import copy_model_files, load_config -from TTS.utils.tensorboard_logger import TensorboardLogger from TTS.utils.training import setup_torch_training_env from TTS.vocoder.datasets.gan_dataset import GANDataset from TTS.vocoder.datasets.preprocess import load_wav_data, load_wav_feat_data @@ -24,11 +26,6 @@ from TTS.vocoder.utils.generic_utils import (plot_results, setup_discriminator, setup_generator) from TTS.vocoder.utils.io import save_best_model, save_checkpoint -# DISTRIBUTED -from torch.nn.parallel import DistributedDataParallel as DDP_th -from torch.utils.data.distributed import DistributedSampler -from TTS.utils.distribute import init_distributed - use_cuda, num_gpus = setup_torch_training_env(True, True)