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

fix random_audio_crop injector

Browse Source
This commit is contained in:
James Betker 2022-03-12 20:42:29 -07:00
parent 8f130e2b3f
commit 08599b4c75
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -74,11 +74,15 @@ class RandomAudioCropInjector(Injector):
def __init__(self, opt, env): def __init__(self, opt, env):
super().__init__(opt, env) super().__init__(opt, env)
self.crop_sz = opt['crop_size'] self.crop_sz = opt['crop_size']
self.lengths_key = opt['lengths_key']
def forward(self, state): def forward(self, state):
inp = state[self.input] inp = state[self.input]
len = inp.shape[-1] lens = state[self.lengths_key]
len = torch.min(lens)
margin = len - self.crop_sz margin = len - self.crop_sz
if margin < 0:
return {self.output: inp}
start = random.randint(0, margin) start = random.randint(0, margin)
return {self.output: inp[:, :, start:start+self.crop_sz]} return {self.output: inp[:, :, start:start+self.crop_sz]}