Tensorflow:如何冻结low_latency_svdf模型图以进行语音识别

时间:2019-01-08 19:18:19

标签: tensorflow

我根据Tensorflow的官方教程训练了low_latency_svdf声学模型:

https://www.tensorflow.org/tutorials/sequences/audio_recognition

训练成功

我尝试在完成训练后冻结图形:

python tensorflow/examples/speech_commands/freeze.py 
--model_architecture=low_latency_svdf 
--start_checkpoint=/tmp/speech_commands_train/low_latency_svdf.ckpt-135000 
--output_file=/tmp/my_frozen_graph.pb

现在的错误是:

“还原中的文件” /Users/michaelhe/ml/env/lib/python2.7/site-packages/tensorflow/python/training/saver.py“,行1582 错误,“当前图和图之间不匹配”) tensorflow.python.framework.errors_impl.InvalidArgumentError:从检查点还原失败。这很可能是由于当前图形与来自检查点的图形之间的不匹配。请确保您没有更改基于检查点的预期图形。原始错误:

分配需要两个张量的形状匹配。 lhs shape = [256,12] rhs shape = [256,16]“

有什么想法吗? 谢谢。

0 个答案:

没有答案