OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Work around protobuf issues by importing onnx first (#6833)

Browse Source
This commit is contained in:
anderspapitto 2018-04-23 15:44:04 -07:00 committed by GitHub
parent aa56a1211d
commit 750a323ca1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
caffe2/python/onnx/backend.py
View File

@ -15,6 +15,13 @@ import collections
from subprocess import Popen, PIPE from subprocess import Popen, PIPE
import zipfile import zipfile
# When onnx is built against a version of protobuf that is older than
# that which is vendored with caffe2, onnx will crash if caffe2's
# vendored protobuf is loaded first. We can work around this by
# importing onnx first, which will cause it to go out and pick up the
# system protobuf.
import onnx.backend
import caffe2 import caffe2
from caffe2.python import core, workspace, rnn_cell, gru_cell from caffe2.python import core, workspace, rnn_cell, gru_cell
from caffe2.python.model_helper import ModelHelper from caffe2.python.model_helper import ModelHelper