Skip to content
Snippets Groups Projects
Commit 6f1285c0 authored by chinyun_yu_joey's avatar chinyun_yu_joey
Browse files

update model(avg_sdr ~= -1.1774)

parent e996a234
No related branches found
No related tags found
No related merge requests found
File added
......@@ -111,7 +111,7 @@ def separate(
class XUMXPredictor(MusicDemixingPredictor):
def prediction_setup(self):
# Load your model here and put it into `evaluation` mode
self.separator = torch.jit.load("./models/xumx_mwf_v3.pth")
self.separator = torch.jit.load("./models/xumx_mwf_v4.pth")
self.separator.eval()
def prediction(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment