try 1 iter of MWF

test_fuser.py

@@ -10,8 +10,8 @@ from evaluator.music_demixing import MusicDemixingPredictor
 
 
 fusion_weights = np.array([0.27203053, 0.73882418])
-fusion_weights_batch = np.array([[0.2, 0.3, 0.5, 0.6],
-                                 [0.8, 0.7, 0.5, 0.4]])
+fusion_weights_batch = np.array([[0.25, 0.27, 0.5, 0.6],
+                                 [0.75, 0.73, 0.5, 0.4]])
 fusion_weights_full = np.array(
     [[0.12421362233786719, 0.053170121998740195, -0.023004887368765208, -0.030820051344791908, 0.7362167118453092, -0.2011487819137626, -0.1254105898553318, -0.11682101700727994, 0.14540460690976179],
      [0.06420106209073237, 0.07177930713127706, 0.0035666737989309845, 0.0005502465774743552, -
@@ -46,7 +46,7 @@ def separate(
     audio,
     unet_model,
     demucs_model,
-    niter=0,
+    niter=1,
     shifts=3
 ):
     unet_predict = unet_separate(audio, unet_model, niter=niter)