From e3eda159d160796efcb4eb21ab2d77cd0e62c707 Mon Sep 17 00:00:00 2001 From: erogol Date: Wed, 25 Nov 2020 14:50:50 +0100 Subject: [PATCH] wavegrad_dataset update --- TTS/vocoder/datasets/wavegrad_dataset.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/TTS/vocoder/datasets/wavegrad_dataset.py b/TTS/vocoder/datasets/wavegrad_dataset.py index 30cf9cb3..22f2af32 100644 --- a/TTS/vocoder/datasets/wavegrad_dataset.py +++ b/TTS/vocoder/datasets/wavegrad_dataset.py @@ -28,7 +28,7 @@ class WaveGradDataset(Dataset): self.ap = ap self.item_list = items - self.seq_len = seq_len + self.seq_len = seq_len if return_segments else None self.hop_len = hop_len self.pad_short = pad_short self.conv_pad = conv_pad @@ -38,7 +38,8 @@ class WaveGradDataset(Dataset): self.use_noise_augment = use_noise_augment self.verbose = verbose - assert seq_len % hop_len == 0, " [!] seq_len has to be a multiple of hop_len." + if return_segments: + assert seq_len % hop_len == 0, " [!] seq_len has to be a multiple of hop_len." self.feat_frame_len = seq_len // hop_len + (2 * conv_pad) # cache acoustic features