wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-17 12:00:07 +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/loader_tests.py
+11 -5
View File
@@ -43,9 +43,10 @@ class TestDataset(unittest.TestCase):
text_lengths = data[1]
linear_input = data[2]
mel_input = data[3]
stop_targets = data[4]
item_idx = data[5]
mel_lengths = data[4]
stop_target = data[5]
item_idx = data[6]
neg_values = text_input[text_input < 0]
check_count = len(neg_values)
assert check_count == 0, \
@@ -82,8 +83,9 @@ class TestDataset(unittest.TestCase):
text_lengths = data[1]
linear_input = data[2]
mel_input = data[3]
stop_target = data[4]
item_idx = data[5]
mel_lengths = data[4]
stop_target = data[5]
item_idx = data[6]
# check the last time step to be zero padded
assert mel_input[0, -1].sum() == 0
@@ -92,6 +94,10 @@ class TestDataset(unittest.TestCase):
assert linear_input[0, -2].sum() != 0
assert stop_target[0, -1] == 1
assert stop_target.sum() == 1
assert len(mel_lengths.shape) == 1
print(mel_lengths)
print(mel_input)
assert mel_lengths[0] == mel_input[0].shape[0]