search

JusperLee / SPMamba

Apache License 2.0
121 stars 15 forks source link

想问下Mamba的训练速度，以及你所使用的机器 #3

Open wyw97 opened 6 months ago

wyw97 commented 6 months ago

我也在使用过Mamba做sound separation相关的工作，然而我发现Mamba替换后，训练速度更慢了。

JusperLee commented 6 months ago

我的速度还是可以的，我用的是8张4090。

XuWink commented 6 months ago

我的速度还是可以的，我用的是8张4090。

你好，请问有预训练模型吗？我的配置太低（一张4090），好像训练不了？

JusperLee commented 6 months ago

你用bf-16就可以了。预训练模型等我参加完ICASSP后整理一下放出来。