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

remove zero-frame in dataloader

Browse Source
This commit is contained in:
Eren Golge
2019-11-19 12:58:54 +01:00
parent ee788bc558
commit 4873620bc2
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/data.py
+3 -2
View File
@@ -24,7 +24,7 @@ def _pad_tensor(x, length):
def prepare_tensor(inputs, out_steps):
max_len = max((x.shape[1] for x in inputs)) + 1 # zero-frame
max_len = max((x.shape[1] for x in inputs))
remainder = max_len % out_steps
pad_len = max_len + (out_steps - remainder) if remainder > 0 else max_len
return np.stack([_pad_tensor(x, pad_len) for x in inputs])
@@ -38,7 +38,8 @@ def _pad_stop_target(x, length):
def prepare_stop_target(inputs, out_steps):
max_len = max((x.shape[0] for x in inputs)) + 1 # zero-frame
""" Pad row vectors with 1. """
max_len = max((x.shape[0] for x in inputs))
remainder = max_len % out_steps
pad_len = max_len + (out_steps - remainder) if remainder > 0 else max_len
return np.stack([_pad_stop_target(x, pad_len) for x in inputs])