Tensorflow Eigen / Freeze_graph问题

时间:2017-04-16 05:31:03

标签: tensorflow deep-learning

我有一个训练有素的模特,我想把它放在我的iPhone上进行推理。我使用了Tensorflow,并获得了必要的graph.pb和检查点文件。我使用freeze_graph.py脚本来冻结带有权重的图形,但是我得到了错误: google.protobuf.text_format.ParseError: 2:1 : Expected identifier or number.在脚本尝试在图表中读取的行。

我尝试使用提到here的方法并安装了bazel。但是,当我执行bazel build tensorflow/python:freeze_graph时,它会给出错误,即在deps中找不到Eigen。这是正确的,因为Eigen在另一个文件夹中。

我理解我在这里要问的两件事:

  • 如果出错,如何正确冻结图表。
  • 如果需要使用bazel来freeze_graph如何绕过Eigen依赖。

我也引用了this问题,但它没有帮助,因为我使用Mac并配置了tensorflow来构建而没有CUDA支持。

任何帮助将不胜感激。感谢。

0 个答案:

没有答案