该博客介绍了如何利用ngraph-bridge的tf_graph_writer工具和TensorFlow的stringDumpGraphToFile方法来缩减模型文件大小。通过将tf_graph.pbtxt文件过滤掉tensor_content，可以生成更小的tf_graph1.pbtxt文件，从而优化模型的存储和加载效率。
https://github.com/tensorflow/ngraph-bridge
ngraph-bridge-0.22.0-rc4\logging\tf_graph_writer.*
直接用tensorflow\core\util\dump_graph.h的 方法
string DumpGraphToFile(const string& name, Graph const& graph,
const FunctionLibraryDefinition* flib_def = nullptr,
const string& dirname = "");
grep -v tensor_content tf_graph.pbtxt > tf_graph1.pbtxt
用于缩减模型文件大小
扫一扫
2万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
