Update inference/mastering_transfer.py

inference/mastering_transfer.py

@@ -85,13 +85,10 @@ class Mastering_Style_Transfer_Inference:
     # Inference whole song
     def inference(self, input_track_path, reference_track_path):
         print("\n======= Start to inference music mastering style transfer =======")
-        # normalized input
-        output_name_tag = 'output' if self.args.normalize_input else 'output_notnormed'
 
+        # load input wavs
         input_aud = load_wav_segment(input_track_path, axis=0)
         reference_aud = load_wav_segment(reference_track_path, axis=0)
-        input_aud = torch.FloatTensor(input_aud).to(self.device)
-        reference_aud = torch.FloatTensor(reference_aud).to(self.device)
 
         # loudness normalization for stability
         meter = pyloudnorm.Meter(44100)
@@ -103,6 +100,9 @@ class Mastering_Style_Transfer_Inference:
         reference_aud = pyloudnorm.normalize.loudness(reference_aud, loudness_ref, -12)
         reference_aud = np.clip(reference_aud, -1., 1.)
         
+        input_aud = torch.FloatTensor(input_aud).to(self.device)
+        reference_aud = torch.FloatTensor(reference_aud).to(self.device)
+        
         cur_out_dir = './yt_dir/0/'
         os.makedirs(cur_out_dir, exist_ok=True)
         ''' segmentize whole songs into batch '''