From 05c75aa9d5358065f86fc321a2c843b0f41add38 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Fri, 16 Jul 2021 13:37:38 +0200 Subject: [PATCH] Fix linter issues --- TTS/trainer.py | 2 -- TTS/vocoder/models/wavegrad.py | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/TTS/trainer.py b/TTS/trainer.py index f3f45ebd..903aee5f 100644 --- a/TTS/trainer.py +++ b/TTS/trainer.py @@ -40,7 +40,6 @@ from TTS.utils.logging import ConsoleLogger, TensorboardLogger from TTS.utils.trainer_utils import get_optimizer, get_scheduler, is_apex_available, setup_torch_training_env from TTS.vocoder.datasets.preprocess import load_wav_data, load_wav_feat_data from TTS.vocoder.models import setup_model as setup_vocoder_model -from TTS.vocoder.models.wavegrad import Wavegrad if platform.system() != "Windows": # https://github.com/pytorch/pytorch/issues/973 @@ -772,7 +771,6 @@ class Trainer: figures, audios = self.model.test_run(self.ap) self.tb_logger.tb_test_audios(self.total_steps_done, audios, self.config.audio["sample_rate"]) self.tb_logger.tb_test_figures(self.total_steps_done, figures) - return None def _fit(self) -> None: """🏃 train -> evaluate -> test for the number of epochs.""" diff --git a/TTS/vocoder/models/wavegrad.py b/TTS/vocoder/models/wavegrad.py index 22d2a015..d2983be2 100644 --- a/TTS/vocoder/models/wavegrad.py +++ b/TTS/vocoder/models/wavegrad.py @@ -273,7 +273,7 @@ class Wavegrad(BaseModel): self.compute_noise_level(betas) for sample in samples: x = sample[0] - x = x[None, : , :].to(next(self.parameters()).device) + x = x[None, :, :].to(next(self.parameters()).device) y = sample[1] y = y[None, :] # compute voice