Dual-Path-RNN-Pytorch icon indicating copy to clipboard operation
Dual-Path-RNN-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon JusperLee

DPRNN batch size

Open ZhangYu-x opened this issue 3 years ago • 6 comments

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

ZhangYu-x avatar Apr 01 '22 13:04 ZhangYu-x

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

JusperLee avatar Apr 02 '22 06:04 JusperLee

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

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

ZhangYu-x avatar Apr 02 '22 08:04 ZhangYu-x

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

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

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

JusperLee avatar Apr 02 '22 08:04 JusperLee

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

ZhangYu-x avatar Apr 02 '22 08:04 ZhangYu-x

https://github.com/JusperLee/Dual-Path-RNN-Pytorch/blob/520ff8d2111bd19a54e5f3b7bde148d80b1eafdb/config/Dual_RNN/train_rnn.yml#L33 这个换成gLN

JusperLee avatar Apr 02 '22 16:04 JusperLee

https://github.com/JusperLee/Dual-Path-RNN-Pytorch/blob/520ff8d2111bd19a54e5f3b7bde148d80b1eafdb/config/Dual_RNN/train_rnn.yml#L33

这个换成gLN

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

ZhangYu-x avatar Apr 03 '22 06:04 ZhangYu-x