Update pretrained_models/2stems/2stems.json

pretrained_models/2stems/2stems.json

@@ -1,28 +1,19 @@
 {
-    "train_csv": "path/to/train.csv",
-    "validation_csv": "path/to/test.csv",
-    "model_dir": "2stems",
-    "mix_name": "mix",
-    "instrument_list": ["vocals", "accompaniment"],
-    "sample_rate":44100,
-    "frame_length":4096,
-    "frame_step":1024,
-    "T":512,
-    "F":1024,
-    "n_channels":2,
-    "separation_exponent":2,
-    "mask_extension":"zeros",
-    "learning_rate": 1e-4,
-    "batch_size":4,
-    "training_cache":"training_cache",
-    "validation_cache":"validation_cache",
-    "train_max_steps": 1000000,
-    "throttle_secs":300,
-    "random_seed":0,
-    "save_checkpoints_steps":150,
-    "save_summary_steps":5,
-    "model":{
-            "type":"unet.unet",
-            "params":{}
-            }
+  "stft_backend": "tensorflow",
+  "mix_name": "mix",
+  "instrumental": {
+    "filters": [
+      {
+        "type": "separate",
+        "left": "voice",
+        "right": "accompaniment"
+      }
+    ]
+  },
+  "voice": {
+    "path": "pretrained_models/2stems/model"
+  },
+  "accompaniment": {
+    "path": "pretrained_models/2stems/model"
+  }
 }