Update inference/style_transfer.py

inference/style_transfer.py

@@ -70,9 +70,6 @@ class Mixing_Style_Transfer_Inference:
                                         num_workers=args.workers, \
                                         drop_last=False)
 
-        # save current arguments
-        self.save_args(args)
-
         ''' check stem-wise result '''
         if not self.args.do_not_separate:
             os.environ['MKL_THREADING_LAYER'] = 'GNU'
@@ -267,8 +264,8 @@ class Mixing_Style_Transfer_Inference:
                     sf.write(os.path.join(cur_out_dir, f"{cur_inst_name}_{output_name_tag}.wav"), fin_data_out_inst.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
             # remix
             fin_data_out_mix = sum(inst_outputs)
-            fin_output_path = os.path.join(cur_out_dir, f"mixture_{output_name_tag}.wav")
-            sf.write(fin_output_path, fin_data_out_mix.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
+            fin_output_path = os.path.join(cur_out_dir, f"mixture_{output_name_tag}.wav"
+            sf.write(fin_output_path), fin_data_out_mix.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
 
             return fin_output_path
 
@@ -304,28 +301,7 @@ class Mixing_Style_Transfer_Inference:
         return whole_batch_data
 
 
-    # save current inference arguments
-    def save_args(self, params):
-        info = '\n[args]\n'
-        for sub_args in parser._action_groups:
-            if sub_args.title in ['positional arguments', 'optional arguments', 'options']:
-                continue
-            size_sub = len(sub_args._group_actions)
-            info += f'  {sub_args.title} ({size_sub})\n'
-            for i, arg in enumerate(sub_args._group_actions):
-                prefix = '-'
-                info += f'      {prefix} {arg.dest:20s}: {getattr(params, arg.dest)}\n'
-        info += '\n'
-
-        os.makedirs(self.output_dir, exist_ok=True)
-        record_path = f"{self.output_dir}style_transfer_inference_configurations.txt"
-        f = open(record_path, 'w')
-        np.savetxt(f, [info], delimiter=" ", fmt="%s")
-        f.close()
-
-
-
-def set_up():
+def set_up()
     os.environ['MASTER_ADDR'] = '127.0.0.1'
     os.environ["CUDA_VISIBLE_DEVICES"] = '0'
     os.environ['MASTER_PORT'] = '8888'