From ae3ecf59073a55be58bb026057d8f6253856eba3 Mon Sep 17 00:00:00 2001 From: antenna Date: Wed, 31 May 2023 02:21:33 +0900 Subject: [PATCH] fix Audio Dataset generate pair's bug --- deepafx_st/data/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepafx_st/data/dataset.py b/deepafx_st/data/dataset.py index 41ebff6..d3617e0 100755 --- a/deepafx_st/data/dataset.py +++ b/deepafx_st/data/dataset.py @@ -237,8 +237,8 @@ def generate_pair(self): )[0] # peak normalize again before passing through dynamic range compressor - input_audio_corrupt /= input_audio_corrupt.abs().max() - input_audio_corrupt *= 10 ** (-12.0 / 20) # with min 3 dBFS headroom + target_audio_corrupt /= target_audio_corrupt.abs().max() + target_audio_corrupt *= 10 ** (-12.0 / 20) # with min 3 dBFS headroom if self.drc_corrupt and torch.rand(1).sum() < 0.75: target_audio_corrupt = augmentations.dynamic_range_compression(