优化冻结图

时间:2017-07-19 18:07:06

标签: python tensorflow

我已经冻结了张量流图,我注意到实时使用图表进行预测太慢了。所以我想做图优化。

在我的项目文件中,我有一个名为model的文件夹,其中包含模型文件(pb文件)。

现在我尝试在模型目录中运行以下命令 bazel build tensorflow/python/tools:strip_unused 但它触发了以下错误,

  

bazel build tensorflow / python / tools:strip_unused   错误:没有这样的包' tensorflow / python / tools':在包路径上找不到BUILD文件。

1 个答案:

答案 0 :(得分:0)

您需要在tensorflow目录中构建工具,而不是模型目录,然后在模型上运行该工具。