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

masked loss

Browse Source
This commit is contained in:
Eren Golge
2018-03-22 13:46:52 -07:00
parent 0f3b2ddd7b
commit 33937f54d0
4 changed files with 39 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/layers_tests.py
+1 -5
View File
@@ -37,16 +37,12 @@ class DecoderTests(unittest.TestCase):
dummy_memory = T.autograd.Variable(T.rand(4, 120, 32))
print(layer)
output, alignment, stop_output = layer(dummy_input, dummy_memory)
output, alignment = layer(dummy_input, dummy_memory)
print(output.shape)
print(" > Stop ", stop_output.shape)
assert output.shape[0] == 4
assert output.shape[1] == 120 / 5
assert output.shape[2] == 32 * 5
assert stop_output.shape[0] == 4
assert stop_output.shape[1] == 120 / 5
assert stop_output.shape[2] == 5
class EncoderTests(unittest.TestCase):