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

Missing keys while loading pre-trained FCOS-3D weights #58

Open
amundra15 opened this issue Oct 24, 2023 · 0 comments
Open

Missing keys while loading pre-trained FCOS-3D weights #58

amundra15 opened this issue Oct 24, 2023 · 0 comments

Comments

@amundra15
Copy link

While loading the pre-trained FCOS-3D weights for training, the code complains about incompatible keys.

Upon analyzing it, most of the "incompatible" keys are as expected (eg, tpv_aggregator, tpv_head related keys missing in the loaded model and bbox_head keys missing in the defined model).

However, one set of items still did not make sense to me. For the image_backbone (and possibly for image_neck) batch norm layers, the running_mean, running_var, and num_batches_tracked keys are present in the checkpoint, but not in my model definition. I am not sure if this is intended -- a quick Google search says that these parameters are critical for training and inference, even if the BN parameters are frozen.

Could you shed some light on this?

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