Fixed default parameters for models

nomadkaraoke · Dec 18, 2024 · 6e86910 · 6e86910
1 parent dcc53e4
commit 6e86910
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/audio_separator/separator/separator.py b/audio_separator/separator/separator.py
@@ -92,10 +92,10 @@ def __init__(
         sample_rate=44100,
         use_soundfile=False,
         use_autocast=False,
-        mdx_params={},
-        vr_params={},
-        demucs_params={},
-        mdxc_params={},
+        mdx_params={"hop_length": 1024, "segment_size": 256, "overlap": 0.25, "batch_size": 1, "enable_denoise": False},
+        vr_params={"batch_size": 1, "window_size": 512, "aggression": 5, "enable_tta": False, "enable_post_process": False, "post_process_threshold": 0.2, "high_end_process": False},
+        demucs_params={"segment_size": "Default", "shifts": 2, "overlap": 0.25, "segments_enabled": True},
+        mdxc_params={"segment_size": 256, "override_model_segment_size": False, "batch_size": 1, "overlap": 8, "pitch_shift": 0},
         info_only=False,
     ):
         """Initialize the separator."""

diff --git a/tests/model-metrics/test-all-models.py b/tests/model-metrics/test-all-models.py
@@ -311,9 +311,7 @@ def main():
                 if combined_results[test_model]["track_scores"]:
                     median_scores = calculate_median_scores(combined_results[test_model]["track_scores"])
                     combined_results[test_model]["median_scores"] = median_scores
-                else:
-                    del combined_results[test_model]
-
+
                 # Save results after each model
                 os.makedirs(os.path.dirname(combined_results_path), exist_ok=True)
                 with open(combined_results_path, "w") as f: