getting unexpected keys error while loading weights

Question

import torch
from PIL import Image
import numpy as np
from effdet import get_efficientdet_config, EfficientDet


config = get_efficientdet_config('tf_efficientdet_d0')
model = EfficientDet(config, pretrained_backbone=True)
model.eval()

when I run this I am getting the error

Unexpected keys (bn2.bias, bn2.num_batches_tracked, bn2.running_mean, bn2.running_var, bn2.weight, classifier.bias, classifier.weight, conv_head.weight) found while loading pretrained weights. This may be expected if model is being adapted.

I researched some bit and got to know that this is because of timm builder but didn't find any solutions. How to fix this?

I wanted to load the efficientdet weights but it resulted in an unexpected keys error

ndrwnaguib · Accepted Answer · 2024-08-12 22:10:01Z

0

Your code works without issues in my environment; I believe your huggingface_hub package is old. Try

python -m pip install --upgrade huggingface_hub

answered Aug 12, 2024 at 22:10

ndrwnaguib

6,2463 gold badges33 silver badges56 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

getting unexpected keys error while loading weights

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related