Skip to content
Snippets Groups Projects
Commit 706098a1 authored by chinyun_yu_joey's avatar chinyun_yu_joey
Browse files

first cfp model (sdr ~= 4.9)

parent 41c7a8ff
No related branches found
Tags submission-demucs-cfp
No related merge requests found
File added
......@@ -12,7 +12,7 @@ def separate(
audio,
rate,
model,
shifts=6,
shifts=5,
device="cpu"
):
......@@ -49,7 +49,7 @@ def separate(
class DemucsPredictor(MusicDemixingPredictor):
def prediction_setup(self):
# Load your model here and put it into `evaluation` mode
self.separator = torch.jit.load('./models/demucs.pth')
self.separator = torch.jit.load('./models/demucs_cfp.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