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

The model and loaded state dict do not match exactly when trying to load pretrained weights #120

Open
Chris-hughes10 opened this issue Aug 15, 2022 · 2 comments

Comments

@Chris-hughes10
Copy link

Hi,

When I am trying to run a training epoch using pretrained weights, I am getting a warning that the model and state dict do not match exactly. As I have not modified anything, I suspect that it is a weight/config mismatch.

Whilst there is an example for convnext_tiny, I am making some assumptions here as I am trying to train convnext_base, the command I am using is:
python tools/train.py configs/convnext/cascade_mask_rcnn_convnext_base_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco_in1k.py --cfg-options model.pretrained=https://dl.fbaipublicfiles.com/convnext/coco/cascade_mask_rcnn_convnext_base_1k_3x.pth

Is this the correct pretrained model for this config?

Thanks!

@wszhengjx
Copy link

have you resolved it ? i meet the same question

@Chris-hughes10
Copy link
Author

Unfortunately I was never able to resolve it, so abandoned investigating it further as it doesn't look actively maintained

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants