Update inference/mastering_transfer.py

inference/mastering_transfer.py

@@ -92,12 +92,13 @@ class Mastering_Style_Transfer_Inference:
 
         # loudness normalization for stability
         meter = pyloudnorm.Meter(44100)
+        norm_loudness_gain = -16.
         loudness_in = meter.integrated_loudness(input_aud.transpose(-1, -2))
         loudness_ref = meter.integrated_loudness(reference_aud.transpose(-1, -2))
 
-        input_aud = pyloudnorm.normalize.loudness(input_aud, loudness_in, -12)
+        input_aud = pyloudnorm.normalize.loudness(input_aud, loudness_in, norm_loudness_gain)
         input_aud = np.clip(input_aud, -1., 1.)
-        reference_aud = pyloudnorm.normalize.loudness(reference_aud, loudness_ref, -12)
+        reference_aud = pyloudnorm.normalize.loudness(reference_aud, loudness_ref, norm_loudness_gain)
         reference_aud = np.clip(reference_aud, -1., 1.)
         
         input_aud = torch.FloatTensor(input_aud).to(self.device)