deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Also do dynamic range compression across mel

Browse Source
This commit is contained in:
James Betker 2021-12-10 20:06:24 -07:00
parent faf55684b8
commit 306274245b
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/trainer/injectors/base_injectors.py
View File

@ -617,6 +617,8 @@ class TorchMelSpectrogramInjector(Injector):
assert len(inp.shape) == 2 assert len(inp.shape) == 2
self.mel_stft = self.mel_stft.to(inp.device) self.mel_stft = self.mel_stft.to(inp.device)
mel = self.mel_stft(inp) mel = self.mel_stft(inp)
# Perform dynamic range compression
mel = torch.log(torch.clamp(mel, min=1e-5))
return {self.output: mel} return {self.output: mel}