提交 f75e11dc 编写于 作者: S SunAhong1993

fix the assert

上级 39a88b1f
...@@ -134,8 +134,7 @@ def caffe2paddle(proto, weight, save_dir, caffe_proto): ...@@ -134,8 +134,7 @@ def caffe2paddle(proto, weight, save_dir, caffe_proto):
if (int(ver_part[0]) == 3 and int(ver_part[1]) >= 6) \ if (int(ver_part[0]) == 3 and int(ver_part[1]) >= 6) \
or (int(ver_part[0]) > 3): or (int(ver_part[0]) > 3):
version_satisfy = True version_satisfy = True
assert version_satisfy, 'The version of protobuf must be \ assert version_satisfy, 'google.protobuf >= 3.6.0 is required'
larger than or equal to 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.
先完成此消息的编辑!
想要评论请 注册