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

Fix stopnet training

Browse Source
This commit is contained in:
Eren Gölge
2021-07-24 11:39:54 +02:00
parent d7a9965389
commit fc0c4600bd
6 changed files with 28 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/data_tests/test_loader.py
+1 -1
View File
@@ -207,7 +207,7 @@ class TestTTSDataset(unittest.TestCase):
assert linear_input[1 - idx, -1].sum() == 0
assert mel_input[1 - idx, -1].sum() == 0
assert stop_target[1, mel_lengths[1] - 1] == 1
assert stop_target[1, mel_lengths[1] :].sum() == 0
assert stop_target[1, mel_lengths[1] :].sum() == stop_target.shape[1] - mel_lengths[1]
assert len(mel_lengths.shape) == 1
# check batch zero-frame conditions (zero-frame disabled)