ONNX加载错误

ONNX加载

import onnx

net = onnx.load("./resnet.onnx")

onnx的__init__.py中调用了protobuf, 出现以下错误

[libprotobuf ERROR google/protobuf/io/coded_stream.cc:207] A protocol message was rejected because it was too big (more than 67108864 bytes).

解决方案

export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python

参考：

libprotobuf: A protocol message was rejected because it was too big (more than 67108864 bytes) 解决方法