search

yeyupiaoling / PaddlePaddle-DeepSpeech

基于PaddlePaddle实现的语音识别，中文语音识别。项目完善，识别效果好。支持Windows，Linux下训练和预测，支持Nvidia Jetson开发板预测。
https://yeyupiaoling.blog.csdn.net/article/details/102904306
Apache License 2.0
650 stars 143 forks source link

启动训练提示显存不足 #81

Closed littlestone0806 closed 2 years ago

littlestone0806 commented 2 years ago

image

littlestone0806 commented 2 years ago

显卡是GTX1050Ti 4G版本的

yeyupiaoling commented 2 years ago

训练，最低要8G，正常要11G以上

yeyupiaoling commented 2 years ago

@littlestone0806 你直接下载我的模型使用就好，可以不用训练

littlestone0806 commented 2 years ago

谢谢作者。如果用您提供的模型，在进行数据准备操作之后，接下来该做哪一步？我在做后续的操作时，总提醒我models文件夹下东西不全

littlestone0806 commented 2 years ago

image 我下一步执行评估时报的错

yeyupiaoling commented 2 years ago

@littlestone0806 你看到日志错误信息了吗？你下载的文件放哪了？

littlestone0806 commented 2 years ago

我把下载的thchs_30.gz放到了download_data目录，然后修改了thchs_30.py的路径，也运行了thchs_30.py，在dataset中生成test、train、npz文件也都没问题，但是执行下一步模型导出时python export_model.py --pretrained_model=./models/param/50.pdparams就提示这个 image

littlestone0806 commented 2 years ago

我看了您的博客，说每训练一轮保存一次模型，模型保存在 DeepSpeech/models/ 目录下，会不会是我没有训练，所以在导出模型时报错？

yeyupiaoling commented 2 years ago

我说的是下载我提供的模型，你下载数据集是不行的。

导出模型命令如下，文档写错了：

python export_model.py --resume_model=./models/param/50.pdparams