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([ self.memory_input = torch.cat([
new_memory, self.memory_input[:, :( new_memory, self.memory_input[:, :(
self.memory_size - self.r) * self.memory_dim].clone() self.memory_size - self.r) * self.memory_dim].clone()
], ], dim=-1)
dim=-1)
else: else:
# memory queue size smaller than number of frames per decoder iter # memory queue size smaller than number of frames per decoder iter
self.memory_input = new_memory[:, :self.memory_size * self.memory_dim] self.memory_input = new_memory[:, :self.memory_size * self.memory_dim]

View File

@ -1,9 +1,8 @@
import unittest import unittest
from utils.text import phonemes from utils.text import phonemes
from collections import Counter
class SymbolsTest(unittest.TestCase): 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))) assert sorted(phonemes) == sorted(list(set(phonemes))), " {} vs {} ".format(len(phonemes), len(set(phonemes)))