Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

UnpicklingError: A load persistent id instruction was encountered, but no persistent_load function was specified. #161

Open
deven-gqc opened this issue Nov 7, 2022 · 0 comments

Comments

@deven-gqc
Copy link

I have a model trained with Pytorch which does instance segmentation, I'm trying to load the model in instanceSegmentation, but I keep getting a UnpicklingError error. When I load that same model with pytorch it works alright. I don't know what is going wrong over here.

image

Loading the model with pytorch works fine.
image

Here is the entire stacktrace

---------------------------------------------------------------------------
UnpicklingError                           Traceback (most recent call last)
[<ipython-input-14-e2c74af97b78>](https://localhost:8080/#) in <module>
----> 1 ins.load_model("/content/drive/MyDrive/models/fastai_tfl_253img.pkl", network_backbone='resnet34')

4 frames
[/usr/local/lib/python3.7/dist-packages/pixellib/torchbackend/instance/__init__.py](https://localhost:8080/#) in load_model(self, model_path, confidence, detection_speed, network_backbone)
     89 
     90         self.cfg.MODEL.WEIGHTS = model_path
---> 91         self.predictor = DefaultPredictor(self.cfg)
     92 
     93 

[/usr/local/lib/python3.7/dist-packages/pixellib/torchbackend/instance/engine/defaultseng.py](https://localhost:8080/#) in __init__(self, cfg)
    290 
    291         checkpointer = DetectionCheckpointer(self.model)
--> 292         checkpointer.load(cfg.MODEL.WEIGHTS)
    293 
    294         self.aug = T.ResizeShortestEdge(

[/usr/local/lib/python3.7/dist-packages/pixellib/torchbackend/instance/checkpoint/detection_checkpoint.py](https://localhost:8080/#) in load(self, path, *args, **kwargs)
     51             if not has_file:
     52                 path = None  # don't load if not readable
---> 53         ret = super().load(path, *args, **kwargs)
     54 
     55         if need_sync:

[/usr/local/lib/python3.7/dist-packages/fvcore/common/checkpoint.py](https://localhost:8080/#) in load(self, path, checkpointables)
    153             assert os.path.isfile(path), "Checkpoint {} not found!".format(path)
    154 
--> 155         checkpoint = self._load_file(path)
    156         incompatible = self._load_model(checkpoint)
    157         if (

[/usr/local/lib/python3.7/dist-packages/pixellib/torchbackend/instance/checkpoint/detection_checkpoint.py](https://localhost:8080/#) in _load_file(self, filename)
     62         if filename.endswith(".pkl"):
     63             with PathManager.open(filename, "rb") as f:
---> 64                 data = pickle.load(f, encoding="latin1")
     65             if "model" in data and "__author__" in data:
     66                 # file is in Detectron2 model zoo format

UnpicklingError: A load persistent id instruction was encountered,
but no persistent_load function was specified.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant