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

Data loader bug fix 2

Browse Source
This commit is contained in:
Eren Golge
2018-03-26 11:07:15 -07:00
parent 3c084177c6
commit 1ff8d6d2b7
4 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/data.py
+4 -4
View File
@@ -17,13 +17,13 @@ def prepare_data(inputs):
def _pad_tensor(x, length):
_pad = 0
assert x.ndim == 2
return np.pad(x, [[0, 0], [0, length - x.shape[1]]], mode='constant', constant_values=_pad)
x = np.pad(x, [[0, 0], [0, length - x.shape[1]]], mode='constant', constant_values=_pad)
return x
def prepare_tensor(inputs, out_steps):
max_len = max((x.shape[1] for x in inputs)) + 1 # zero-frame
remainder = max_len % out_steps
return np.stack([_pad_tensor(x, max_len + remainder) for x in inputs])
return np.stack([_pad_tensor(x, max_len + (out_steps - remainder)) for x in inputs])
def _pad_stop_target(x, length):
@@ -35,7 +35,7 @@ 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
remainder = max_len % out_steps
return np.stack([_pad_stop_target(x, max_len + remainder) for x in inputs])
return np.stack([_pad_stop_target(x, max_len + (out_steps - remainder)) for x in inputs])
def pad_per_step(inputs, pad_len):