Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to train model for more than 2 speakers? #19

Open
Sonish-Maharjan-2014 opened this issue Jan 3, 2024 · 4 comments
Open

How to train model for more than 2 speakers? #19

Sonish-Maharjan-2014 opened this issue Jan 3, 2024 · 4 comments

Comments

@Sonish-Maharjan-2014
Copy link

I trained the model (form NBSS) branch for 2 speakers separation using wsj0 dataset. It perfectly worked. But now I want to train the model for more than 2 speakers. What steps should I follow?

@quancs
Copy link
Member

quancs commented Jan 11, 2024

Hello, thank you for your insterests in our works. To train the model for a dataset where each utterance more than 2 speakers, you can change the number of output channels to 2N (N speakers and each has 2 numbers for the real and imginary parts of STFT coefficients) for each TF-bin.

@Sonish-Maharjan-2014
Copy link
Author

Thank you for your response.. I tried adapting the code for four speakers. I generated room impulse responses (RIR) for the four speakers and made some adjustments in the code. Unfortunately, I ran into an error towards the end of the process.
image
Could you help me fix the problem?

@quancs
Copy link
Member

quancs commented Jan 26, 2024

You can debug your code to check the shape of echoics, echoic_i, and the value of needed_lens

@Sonish-Maharjan-2014
Copy link
Author

Thanks, I will try to debug

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants