2 stems instead of 4

[coincoin73]

First, thanks for this implementation of improved TasNet model.
Now the output produces 4 stems : bass, drum, other and vocal.
How can I modify it to achieve 2 stems : accompaniment and vocal only ?

Thanks in advance.

[davda54]

Hi, thanks for your interest in our work! There's unfortunately not any trivial way of changing the code for 2 stems (it's pretty much hardcoded for musdb), but I believe it shouldn't be that complicated either. Replacing all occurrences of 4 by 2 should basically do the job :) Let me know if you encounter any problems while converting it.

[coincoin73]

H, I have done the job and it seems to work.
I have just tested on very very small dataset to ensure that all is correct.
I must do the training on the full musdb18.

[davda54]

Hi, that's great! Would you consider making a pull request? :)

[coincoin73]

Hi
Impossible to do the complete training as I only have 1 GPU with 16 GB