mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tacotron stft - loosen bounds restrictions and clip

Browse Source
This commit is contained in:
James Betker 2022-03-15 10:31:26 -06:00
parent f8631ad4f7
commit 9767260c6c
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/models/tacotron2/layers.py
View File

@ -70,8 +70,9 @@ class TacotronSTFT(torch.nn.Module):
------- -------
mel_output: torch.FloatTensor of shape (B, n_mel_channels, T) mel_output: torch.FloatTensor of shape (B, n_mel_channels, T)
""" """
assert(torch.min(y.data) >= -1) assert(torch.min(y.data) >= -10)
assert(torch.max(y.data) <= 1) assert(torch.max(y.data) <= 10)
y = torch.clip(y, min=-1, max=1)
magnitudes, phases = self.stft_fn.transform(y) magnitudes, phases = self.stft_fn.transform(y)
magnitudes = magnitudes.data magnitudes = magnitudes.data