RuntimeError: Error(s) in loading state_dict for FasterRCNN

I’ve saved a model using model.save('model_weights.pth') after a successful run.

When now attempting to load the model using the code;

from detecto.core import Model

model = Model.load('model_weights.pth', ['aboriginalflags'])

I get the error

Traceback (most recent call last):
  File "show.py", line 5, in <module>
    model = Model.load('model_weights.pth', ['aboriginalflags'])
  File "/usr/local/lib/python3.8/dist-packages/detecto/core.py", line 566, in load
    model._model.load_state_dict(torch.load(file, map_location=model._device))
  File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 1044, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for FasterRCNN:
	size mismatch for roi_heads.box_predictor.cls_score.weight: copying a param with shape torch.Size([3, 1024]) from checkpoint, the shape in current model is torch.Size([2, 1024]).
	size mismatch for roi_heads.box_predictor.cls_score.bias: copying a param with shape torch.Size([3]) from checkpoint, the shape in current model is torch.Size([2]).
	size mismatch for roi_heads.box_predictor.bbox_pred.weight: copying a param with shape torch.Size([12, 1024]) from checkpoint, the shape in current model is torch.Size([8, 1024]).
	size mismatch for roi_heads.box_predictor.bbox_pred.bias: copying a param with shape torch.Size([12]) from checkpoint, the shape in current model is torch.Size([8]).

Ubuntu 20

(loving the lib, great work)

[smithajo]

Were you able to solve the issue ? Could you share how you did it?

[SutirthaChakraborty]

Is it still the problem ?