fix travis + pylint tests

This commit is contained in:
sanjaesc 2020-10-22 13:22:50 +02:00 committed by erogol
parent 91e5f8b63d
commit 7c72562fe7
2 changed files with 3 additions and 2 deletions

View File

@ -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

View File

@ -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)