那棵树看起来生气了
[常见错误]TensorFlow模型量化protobuf大小限制
12/19
前言
在做TensorFlow模型量化过程中，会用到transform_graph，然后在进行模型压缩时候因为自己的模型有1.7GB，所以程序报错。
[libprotobuf ERROR external/protobuf_archive/src/google/protobuf/io/coded_stream.cc:190] A protocol message was rejected because it was too big (more than 1073741824 bytes). To increase the limit (or to disable these warnings), see CodedInputStream::SetTotalBytesLimit() in google/protobuf/io/coded_stream.h.
解决办法
- 修改代码
$ vim tensorflow/core/platform/env.cc
找到coded_stream.SetTotalBytesLimit(1024LL << 20, 512LL << 20);
coded_stream.SetTotalBytesLimit(INT_MAX, INT_MAX);然后修改参数即可
- 重新编译
$ bazel build tensorflow/tools/graph_transforms:transform_graph
三合一收款
下面三种方式都支持哦
