From 7c72562fe779261fec4de4161d02a05c7babaa12 Mon Sep 17 00:00:00 2001 From: sanjaesc Date: Thu, 22 Oct 2020 13:22:50 +0200 Subject: [PATCH] fix travis + pylint tests --- .travis/script | 3 ++- TTS/vocoder/models/wavernn.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.travis/script b/.travis/script index 0c24a221..0860f9cf 100755 --- a/.travis/script +++ b/.travis/script @@ -17,5 +17,6 @@ fi if [[ "$TEST_SUITE" == "testscripts" ]]; then # test model training scripts ./tests/test_tts_train.sh - ./tests/test_vocoder_train.sh + ./tests/test_vocoder_gan_train.sh + ./tests/test_vocoder_wavernn_train.sh fi diff --git a/TTS/vocoder/models/wavernn.py b/TTS/vocoder/models/wavernn.py index 9b151cac..8a45d9e3 100644 --- a/TTS/vocoder/models/wavernn.py +++ b/TTS/vocoder/models/wavernn.py @@ -225,7 +225,7 @@ class WaveRNN(nn.Module): def forward(self, x, mels): bsize = x.size(0) - h1 = torch.zeros(1, bsize, self.rnn_dims).to(x.device) + h1 = torch.zeros(1, bsize, self.rnn_dims).to(x.device) h2 = torch.zeros(1, bsize, self.rnn_dims).to(x.device) mels, aux = self.upsample(mels)