kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

channel clipper

Browse Source
This commit is contained in:
James Betker 2022-06-13 20:37:35 -06:00
parent 47330d603b
commit 7ff1fbe2be
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -368,5 +368,16 @@ class NormalizeMelInjector(Injector):
return {self.output: normalize_mel(mel)} return {self.output: normalize_mel(mel)}
class ChannelClipInjector(Injector):
def __init__(self, opt, env):
super().__init__(opt, env)
self.lo = opt['channel_low']
self.hi = opt['channel_high']
def forward(self, state):
inp = state[self.input]
return {self.output: inp[:,self.lo:self.hi]}
if __name__ == '__main__': if __name__ == '__main__':
print('hi') print('hi')