Update inference/mastering_transfer.py

inference/mastering_transfer.py

@@ -90,19 +90,19 @@ class Mastering_Style_Transfer_Inference:
         input_aud = load_wav_segment(input_track_path, axis=0)
         reference_aud = load_wav_segment(reference_track_path, axis=0)
 
-        cur_out_dir = dir_name[0].replace(self.target_dir, self.output_dir)
+        cur_out_dir = './yt_dir/0/'
         os.makedirs(cur_out_dir, exist_ok=True)
         ''' segmentize whole songs into batch '''
         if input_aud.shape[1] > self.args.segment_length:
             cur_inst_input_stem = self.batchwise_segmentization(input_aud, \
-                                                                    dir_name[0], \
+                                                                    "input", \
                                                                     segment_length=self.args.segment_length, \
                                                                     discard_last=False)
         else:
             cur_inst_input_stem = [input_aud.unsqueeze(0)]
         if reference_aud.shape[1] > self.args.segment_length*2:
             cur_inst_reference_stem = self.batchwise_segmentization(reference_aud, \
-                                                                        dir_name[0], \
+                                                                        "reference", \
                                                                         segment_length=self.args.segment_length_ref, \
                                                                         discard_last=False)
         else: