Skip to content
This repository has been archived by the owner on Nov 21, 2023. It is now read-only.

Fix convert_pkl_to_pb import error #976

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Fix convert_pkl_to_pb import error #976

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
2 changes: 1 addition & 1 deletion tools/convert_pkl_to_pb.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down