Update inference/mastering_transfer.py

inference/mastering_transfer.py

@@ -95,8 +95,8 @@ class Mastering_Style_Transfer_Inference:
 
         # loudness normalization for stability
         meter = pyloudnorm.Meter(44100)
-        loudness_in = meter.integrated_loudness(input_aud.transpose())
-        loudness_ref = meter.integrated_loudness(reference_aud.transpose())
+        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 = np.clip(input_aud, -1., 1.)
@@ -152,7 +152,7 @@ class Mastering_Style_Transfer_Inference:
         fin_data_out_mastered = fin_data_out[:, :input_aud.shape[-1]].numpy()
 
         # adjust to reference's loudness
-        loudness_out = meter.integrated_loudness(fin_data_out_mastered.transpose())
+        loudness_out = meter.integrated_loudness(fin_data_out_mastered.transpose(-1, -2))
         fin_data_out_mastered = pyloudnorm.normalize.loudness(fin_data_out_mastered, loudness_out, loudness_ref)
         fin_data_out_mastered = np.clip(fin_data_out_mastered, -1., 1.)