From 4157e99d2d9bb5c87b09fbfb3be241f43a0079c6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Thu, 30 Sep 2021 14:38:34 +0000 Subject: [PATCH] Allow custom padding value --- TTS/tts/utils/data.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/TTS/tts/utils/data.py b/TTS/tts/utils/data.py index b0d88740..09f3cad7 100644 --- a/TTS/tts/utils/data.py +++ b/TTS/tts/utils/data.py @@ -1,15 +1,14 @@ import numpy as np -def _pad_data(x, length): - _pad = 0 +def _pad_data(x, length, pad_val=0): assert x.ndim == 1 - return np.pad(x, (0, length - x.shape[0]), mode="constant", constant_values=_pad) + return np.pad(x, (0, length - x.shape[0]), mode="constant", constant_values=pad_val) -def prepare_data(inputs): +def prepare_data(inputs, pad_val=0): max_len = max((len(x) for x in inputs)) - return np.stack([_pad_data(x, max_len) for x in inputs]) + return np.stack([_pad_data(x, max_len, pad_val) for x in inputs]) def _pad_tensor(x, length):