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

DPRNN batch size #46

Open
ZhangYu-x opened this issue Apr 1, 2022 · 6 comments
Open

DPRNN batch size #46

ZhangYu-x opened this issue Apr 1, 2022 · 6 comments

Comments

@ZhangYu-x
Copy link

您好,我看到您的知乎分享贴中说DPRNN好像一个batch size效果最好, 您试过batch size等于2或者更高的时候吗,您在代码的readme中说100个epoch之后DPRNN的sisnr能到达18.98dB,那batch size = 2的时候sisnr能到达多少您记得吗?因为batchsize=1训练起来非常慢,我想试试batch size=2的时候,但是不知道能获得什么效果。

@JusperLee
Copy link
Owner

没有试过,因为我的显存不支持batch=2。您可以尝试一下,可能没有什么问题

@ZhangYu-x
Copy link
Author

没有试过,因为我的显存不支持batch=2。您可以尝试一下,可能没有什么问题

sisnr=18.98dB是您在这套DPRNN代码上跑出来的吗,我想确认一下是不是我参数配置的问题,因为我用您的代码最终没能跑到18.98dB。

@JusperLee
Copy link
Owner

没有试过,因为我的显存不支持batch=2。您可以尝试一下,可能没有什么问题

sisnr=18.98dB是您在这套DPRNN代码上跑出来的吗,我想确认一下是不是我参数配置的问题,因为我用您的代码最终没能跑到18.98dB。

是在这个代码中跑出来的。

@ZhangYu-x
Copy link
Author

您的超参数设置和论文里面的参数设置一样吗,我不太明白为什么一样的代码,数据集也是WSJ0-2mix,我跑了50个epoch之后才达到15dB。是RNN模型训练结果有很大的随机性吗?

@JusperLee
Copy link
Owner


这个换成gLN

@ZhangYu-x
Copy link
Author

这个换成gLN

好的,谢谢您,我试一试。

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