[Bug] KeyError: 'message_hub'

[Madeline-hyh]

Prerequisite

• [X] I have searched Issues and Discussions but cannot get the expected help.
• [X] I have read the FAQ documentation but cannot get the expected help.
• [X] The bug has not been fixed in the latest version (dev-1.x) or latest version (dev-1.0).

Task

I'm using the official example scripts/configs for the officially supported tasks/models/datasets.

Branch

1.x branch https://github.com/open-mmlab/mmdetection3d/tree/dev-1.x

Environment

mmcv 2.0.1 mmdet 3.1.0 mmdet3d 1.2.0 mmengine 0.9.0 mmsegmentation 1.2.0

Reproduces the problem - code sample

self.message_hub.load_state_dict(checkpoint['message_hub']) KeyError: 'message_hub'

Reproduces the problem - command or script

self.message_hub.load_state_dict(checkpoint['message_hub']) KeyError: 'message_hub'

Reproduces the problem - error message

self.message_hub.load_state_dict(checkpoint['message_hub']) KeyError: 'message_hub'

Additional information

No response

[ElviraLL]

I think this is because that the checkpoint you load doesn't have message_hub created.

[mmayank74567]

Any solution for this? It looks like pretrained models did not save the message_hub key when saving the model.