search

ZFTurbo / Music-Source-Separation-Training

Repository for training models for music source separation.
369 stars 49 forks source link

Roformer for more than two stems? #56

Open rimb05 opened 1 month ago

rimb05 commented 1 month ago

Is roformer only configured for two stems (for example vocal and other)? I tried to run it with 4 stems (vocal / bass / guitar / drums) but it failed with a tensor size mismatch. Is there a config setting I'm missing? Thanks

ZFTurbo commented 3 weeks ago

Sorry for long reply. I added example here: https://github.com/ZFTurbo/Music-Source-Separation-Training/blob/main/configs/config_musdb18_mel_band_roformer_all_stems.yaml

In short to train for all stems you need to set num_stems: 4. Set instruments: - drums - bass - other - vocals and set target_instrument: null.

I didn't check it yet myself for correctness. But training process started normally.