我有一个训练有素的模特,我想把它放在我的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在另一个文件夹中。
我理解我在这里要问的两件事:
freeze_graph
如何绕过Eigen依赖。我也引用了this问题,但它没有帮助,因为我使用Mac并配置了tensorflow来构建而没有CUDA支持。
任何帮助将不胜感激。感谢。