提交 7db28fdf 编写于 作者: S SunAhong1993

check protobuf

上级 8e130a93
...@@ -124,8 +124,8 @@ def caffe2paddle(proto, weight, save_dir, caffe_proto): ...@@ -124,8 +124,8 @@ def caffe2paddle(proto, weight, save_dir, caffe_proto):
from x2paddle.optimizer.caffe_optimizer import CaffeOptimizer from x2paddle.optimizer.caffe_optimizer import CaffeOptimizer
import google.protobuf as gpb import google.protobuf as gpb
ver_str = gpb.__version__.replace('.', '') ver_str = gpb.__version__.replace('.', '')
ver_int = int(ver_str) ver_int = int(ver_str[0:2])
assert ver_int >= 360, 'The version of protobuf must be larger than 3.6.0!' assert ver_int >= 36, 'The version of protobuf must be larger than 3.6.0!'
print("Now translating model from caffe to paddle.") print("Now translating model from caffe to paddle.")
model = CaffeDecoder(proto, weight, caffe_proto) model = CaffeDecoder(proto, weight, caffe_proto)
mapper = CaffeOpMapper(model) mapper = CaffeOpMapper(model)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册