Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

wavegrad_dataset update

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
Show Stats 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