diff options
Diffstat (limited to 'become_yukarin/acoustic_converter.py')
| -rw-r--r-- | become_yukarin/acoustic_converter.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/become_yukarin/acoustic_converter.py b/become_yukarin/acoustic_converter.py index 498bdb1..62eacff 100644 --- a/become_yukarin/acoustic_converter.py +++ b/become_yukarin/acoustic_converter.py @@ -73,6 +73,9 @@ class AcousticConverter(object): input = self._feature_normalize(input, test=True) input = self._encode_feature(input, test=True) + pad = 128 - input.shape[1] % 128 + input = numpy.pad(input, [(0, 0), (0, pad)], mode='minimum') + converter = partial(chainer.dataset.convert.concat_examples, device=self.gpu, padding=0) inputs = converter([input]) @@ -81,6 +84,7 @@ class AcousticConverter(object): if self.gpu is not None: out = chainer.cuda.to_cpu(out) + out = out[:, :-pad] out = self._decode_feature(out, test=True) out = AcousticFeature( |