search

yeyupiaoling / MASR

Pytorch实现的流式与非流式的自动语音识别框架，同时兼容在线和离线识别，目前支持Conformer、Squeezeformer、DeepSpeech2模型，支持多种数据增强方法。
Apache License 2.0
597 stars 106 forks source link

报错了，debug来看，发现是ctc loss出问题了 #19

Closed gouqi666 closed 2 years ago

gouqi666 commented 2 years ago

image

yeyupiaoling commented 2 years ago

是不是显存不足，你的显存是多少，如果不够的话，设计更小的batch_size

gouqi666 commented 2 years ago

博主大大，不是显存不足啊，我都调到2了，还是不行，不知道是啥问题。全按照你的步骤来的，求解答

gouqi666 commented 2 years ago

能不能加个联系方式，我搞了好久了。。。。

yeyupiaoling commented 2 years ago

你的显存是多大？

yeyupiaoling commented 2 years ago

Pytorch版本是多少，有没有对应上文档的版本

gouqi666 commented 2 years ago

显存还有4G，torch版本是1.10.0，cuda版本是10.0，就这个不一样

yeyupiaoling commented 2 years ago

要不要你按照torch 1.6.0版本试试？4G有点少，一般都要8G以上

gouqi666 commented 2 years ago

啊啊啊啊啊啊啊啊是torch的问题还是cuda的问题啊

gouqi666 commented 2 years ago

我用pytorch的ctcloss能跑，但是loss是负数，天呐

gouqi666 commented 2 years ago

博主大大有提供的现成的模型吗，我这边训练不了而且太久了...

yeyupiaoling commented 2 years ago

都不看文档的吗？文档最后不是有提供吗？

yeyupiaoling commented 2 years ago

发布了最新版本。在操场loss是用torch的，应该没有问题了