Merge pull request #3 from TE-StefanUhlich/fix_eval

Fix evaluation
AIcrowd · May 7, 2021 · 4cd3be8 · 4cd3be8
2 parents 73dc139 + d370a87
commit 4cd3be8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evaluator/music_demixing.py b/evaluator/music_demixing.py
@@ -91,7 +91,7 @@ def sdr(references, estimates):
             estimates = []
             for instrument in instruments:
                 reference_file = join(self.test_data_path, music_name, instrument + ".wav")
-                estimate_file = self.get_music_file_location(music_name, "bass")
+                estimate_file = self.get_music_file_location(music_name, instrument)
                 reference, _ = sf.read(reference_file)
                 estimate, _ = sf.read(estimate_file)
                 references.append(reference)