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

no grads for mel injectors

Browse Source
This commit is contained in:
James Betker 2022-05-23 10:34:53 -06:00
parent dc471f5c6d
commit 36dd4eb61f
1 changed files with 15 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/trainer/injectors/audio_injectors.py
View File

@ -69,6 +69,7 @@ class TorchMelSpectrogramInjector(Injector):
self.mel_norms = None self.mel_norms = None
def forward(self, state): def forward(self, state):
with torch.no_grad():
inp = state[self.input] inp = state[self.input]
if len(inp.shape) == 3: # Automatically squeeze out the channels dimension if it is present (assuming mono-audio) if len(inp.shape) == 3: # Automatically squeeze out the channels dimension if it is present (assuming mono-audio)
inp = inp.squeeze(1) inp = inp.squeeze(1)