This commit is contained in:
Eren Golge 2019-08-16 15:49:12 +02:00
parent e1b3c41af5
commit c637aa04a2
3 changed files with 4 additions and 6 deletions

View File

@ -399,8 +399,7 @@ class Decoder(nn.Module):
self.memory_input = torch.cat([
new_memory, self.memory_input[:, :(
self.memory_size - self.r) * self.memory_dim].clone()
],
dim=-1)
], dim=-1)
else:
# memory queue size smaller than number of frames per decoder iter
self.memory_input = new_memory[:, :self.memory_size * self.memory_dim]

View File

@ -1,9 +1,8 @@
import unittest
from utils.text import phonemes
from collections import Counter
class SymbolsTest(unittest.TestCase):
def test_uniqueness(self):
def test_uniqueness(self): #pylint: disable=no-self-use
assert sorted(phonemes) == sorted(list(set(phonemes))), " {} vs {} ".format(len(phonemes), len(set(phonemes)))

View File

@ -138,8 +138,8 @@ class TestTTSDataset(unittest.TestCase):
# there is a slight difference between two matrices.
# TODO: Check this assert cond more in detail.
assert abs((abs(mel.T)
- abs(mel_dl[:-1])
).sum()) < 1e-5, (abs(mel.T)- abs(mel_dl[:-1])).sum()
- abs(mel_dl[:-1])
).sum()) < 1e-5, (abs(mel.T) - abs(mel_dl[:-1])).sum()
# check mel-spec correctness
mel_spec = mel_input[0].cpu().numpy()