wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-11 12:00:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

docstrings for Tacotron models

Browse Source
This commit is contained in:
erogol
2020-07-23 16:26:20 +02:00
parent 4c7c4f77f8
commit cb4313dcb6
3 changed files with 183 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_layers.py
+5 -5
View File
@@ -31,7 +31,7 @@ class CBHGTests(unittest.TestCase):
gru_features=80,
num_highways=4)
# B x D x T
dummy_input = T.rand(4, 128, 8)
dummy_input = T.rand(4, 128, 8)
print(layer)
output = layer(dummy_input)
@@ -44,8 +44,8 @@ class DecoderTests(unittest.TestCase):
@staticmethod
def test_in_out():
layer = Decoder(
in_features=256,
memory_dim=80,
in_channels=256,
frame_channels=80,
r=2,
memory_size=4,
attn_windowing=False,
@@ -74,8 +74,8 @@ class DecoderTests(unittest.TestCase):
@staticmethod
def test_in_out_multispeaker():
layer = Decoder(
in_features=256,
memory_dim=80,
in_channels=256,
frame_channels=80,
r=2,
memory_size=4,
attn_windowing=False,