You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
Loading the model with pytorch works fine.
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 functionwas specified.
The text was updated successfully, but these errors were encountered:
I have a model trained with Pytorch which does instance segmentation, I'm trying to load the model in
instanceSegmentation
, but I keep getting aUnpicklingError
error. When I load that same model with pytorch it works alright. I don't know what is going wrong over here.Loading the model with pytorch works fine.
Here is the entire stacktrace
The text was updated successfully, but these errors were encountered: