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

Key Error in pretrained model #4

Open
Fly-dream12 opened this issue Jan 4, 2020 · 2 comments
Open

Key Error in pretrained model #4

Fly-dream12 opened this issue Jan 4, 2020 · 2 comments

Comments

@Fly-dream12
Copy link

Thanks for your project!
I have downloaded the pretrained model diou.pth , but when I use this it runs into an error like this:

File "DIoU-pytorch-detectron-master/lib/modeling/model_builder.py", line 128, in _init_modules
resnet_utils.load_pretrained_imagenet_weights(self)
File "DIoU-pytorch-detectron-master/lib/utils/resnet_weights_helper.py", line 38, in load_pretrained_imagenet_weights
bn_mean = pretrianed_state_dict[pretrianed_name + '.running_mean']
KeyError: 'res1.bn1.running_mean'

I wonder if i have not set the right defualt settings

@Zzh-tju
Copy link
Owner

Zzh-tju commented Jan 4, 2020

Have you try other weights?Or maybe environment problem.

@Fly-dream12
Copy link
Author

I have tried to set LOAD_IMAGENET_PRETRAINED_WEIGHTS to False and it can succesfully train.
Is it OK?

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

2 participants