mirror of https://github.com/coqui-ai/TTS.git
wavegrad_dataset update
This commit is contained in:
parent
a1e4ee18f9
commit
e3eda159d1
|
@ -28,7 +28,7 @@ class WaveGradDataset(Dataset):
|
||||||
|
|
||||||
self.ap = ap
|
self.ap = ap
|
||||||
self.item_list = items
|
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.hop_len = hop_len
|
||||||
self.pad_short = pad_short
|
self.pad_short = pad_short
|
||||||
self.conv_pad = conv_pad
|
self.conv_pad = conv_pad
|
||||||
|
@ -38,7 +38,8 @@ class WaveGradDataset(Dataset):
|
||||||
self.use_noise_augment = use_noise_augment
|
self.use_noise_augment = use_noise_augment
|
||||||
self.verbose = verbose
|
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)
|
self.feat_frame_len = seq_len // hop_len + (2 * conv_pad)
|
||||||
|
|
||||||
# cache acoustic features
|
# cache acoustic features
|
||||||
|
|
Loading…
Reference in New Issue