open-unmix-pytorch icon indicating copy to clipboard operation
open-unmix-pytorch copied to clipboard

Zenodo model should have the separator.json file included

Open sevagh opened this issue 3 years ago • 3 comments

If I want to download the umx pretrained model, I can get the target.json and target.pth files from Zenodo, but it's lacking the separator.json: https://zenodo.org/record/3340804

It's pretty easy to generate but do you think this should be in the Zenodo collection of files? FWIW here's the default separator.json for umxhq:

{
  "nfft": 4096,
  "nhop": 1024,
  "sample_rate": 44100,
  "nb_channels": 2
}

sevagh avatar Jul 30 '21 00:07 sevagh

@sevagh yes, good point. Zenodo is painful when it comes to json since you can't download them directly or save them in the root dir... but i can update the repos

faroit avatar Jul 30 '21 06:07 faroit

Can the pretrained model download be a single zip containing the 4 training logs, separator.json, and the 4 pth files? The working UMX setup.

Or is it split up in Zenodo for flexiblity? E.g. to be able to download each model separatey.

sevagh avatar Jul 30 '21 10:07 sevagh

it was split to allow torch.hub to be used with zenodo. For torch.hub a direct link to the pth file is required

faroit avatar Jul 30 '21 19:07 faroit