Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2025-12-25 12:49:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

wavegrad_dataset update

Browse Source
This commit is contained in:
erogol
2020-11-25 14:50:50 +01:00
parent a1e4ee18f9
commit e3eda159d1
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/vocoder/datasets/wavegrad_dataset.py
View File

@@ -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