From ed7d79be8cd362803f86afae8c89c861602f6a82 Mon Sep 17 00:00:00 2001 From: Photon <1764252734@qq.com> Date: Wed, 22 Jan 2020 15:47:56 +0800 Subject: [PATCH] Fix convert_pkl_to_pb import error --- tools/convert_pkl_to_pb.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/convert_pkl_to_pb.py b/tools/convert_pkl_to_pb.py index 442084d89..657cae124 100644 --- a/tools/convert_pkl_to_pb.py +++ b/tools/convert_pkl_to_pb.py @@ -41,7 +41,7 @@ import detectron.utils.model_convert_utils as mutils import detectron.utils.vis as vis_utils import numpy as np -from caffe2.caffe2.fb.predictor import predictor_exporter, predictor_py_utils +from caffe2.python.predictor import predictor_exporter, predictor_py_utils from caffe2.proto import caffe2_pb2 from caffe2.python import core, workspace from caffe2.python.predictor_constants import predictor_constants as predictor_constants