a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests 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.min_crop_sz = opt['crop_size']
self.max_crop_sz = self.min_crop_sz self.max_crop_sz = self.min_crop_sz
else: else:
self.min_crop_sz = opt['min_crop_sz'] self.min_crop_sz = opt['min_crop_size']
self.max_crop_sz = opt['max_crop_sz'] self.max_crop_sz = opt['max_crop_size']
self.lengths_key = opt['lengths_key'] self.lengths_key = opt['lengths_key']
def forward(self, state): def forward(self, state):
@ -106,7 +106,7 @@ class RandomAudioCropInjector(Injector):
if margin < 0: if margin < 0:
return {self.output: inp} 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+crop_sz]}
class AudioClipInjector(Injector): class AudioClipInjector(Injector):