From fcfb3a152526f6b9b36ecc3fd0010ea5f8f1db23 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 21 Jun 2022 20:09:59 -0600 Subject: [PATCH] fix --- codes/trainer/injectors/audio_injectors.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/codes/trainer/injectors/audio_injectors.py b/codes/trainer/injectors/audio_injectors.py index 3c8e29cc..d7b1ab5f 100644 --- a/codes/trainer/injectors/audio_injectors.py +++ b/codes/trainer/injectors/audio_injectors.py @@ -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):