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

fix

Browse Source
This commit is contained in:
James Betker 2022-06-21 20:09:59 -06:00
parent 24e60bd510
commit fcfb3a1525
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -93,8 +93,8 @@ class RandomAudioCropInjector(Injector):
self.min_crop_sz = opt['crop_size']
self.max_crop_sz = self.min_crop_sz
else:
self.min_crop_sz = opt['min_crop_sz']
self.max_crop_sz = opt['max_crop_sz']
self.min_crop_sz = opt['min_crop_size']
self.max_crop_sz = opt['max_crop_size']
self.lengths_key = opt['lengths_key']
def forward(self, state):
@ -106,7 +106,7 @@ class RandomAudioCropInjector(Injector):
if margin < 0:
return {self.output: inp}
start = random.randint(0, margin)
return {self.output: inp[:, :, start:start+self.crop_sz]}
return {self.output: inp[:, :, start:start+crop_sz]}
class AudioClipInjector(Injector):