【发布时间】:2017-12-29 13:05:38
【问题描述】:
我有一个很菜鸟的问题。 我在 gpu 上训练 deepspeech 模型,检查点保存到文件夹“检查点”中。 如果我中断训练过程,我如何使用模型的检查点进行预测? 例如,我想查看模型对某个 wav 文件的预测。 我需要生成描述这个 wav 文件的 csv 文件吗?以及如何检查模型?
非常感谢您的关注,我将不胜感激任何帮助或链接。
【问题讨论】:
标签: python mozilla speech-to-text
我有一个很菜鸟的问题。 我在 gpu 上训练 deepspeech 模型,检查点保存到文件夹“检查点”中。 如果我中断训练过程,我如何使用模型的检查点进行预测? 例如,我想查看模型对某个 wav 文件的预测。 我需要生成描述这个 wav 文件的 csv 文件吗?以及如何检查模型?
非常感谢您的关注,我将不胜感激任何帮助或链接。
【问题讨论】:
标签: python mozilla speech-to-text
我找到了解决方案。指定 wav 文件路径的参数 --one_shot_infer 正是我所需要的。
【讨论】:
我相信您已经使用--export_dir 将模型导出到某个地方。如果你这样做了,那么你可能会找到一个扩展名为 .pb 或 .pbmm 的文件。使用它您可以执行推理。更多详情请访问官方文档here。
您可以从 linux 或 mac 使用以下命令执行推理。
$ deepspeech --model /path/to/model/ --audio /path/to/audio/ --alphabet /path/to/alphabet/ --lm /path/to/langauge/model/ --trie /path/to/trie/file/ 。
要执行 python 推理,请参考同一项目 repo 中的client.py。
【讨论】: