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

Pretrained weights such as [Faster RCNN + GIoU] you supported can't use #10

Open
yunxi1 opened this issue Apr 12, 2021 · 0 comments
Open

Comments

@yunxi1
Copy link

yunxi1 commented Apr 12, 2021

Hi,I download your trained model faster_giou.pth(333mb),and run
python tools/test_net.py --dataset coco2017 --cfg configs/baselines/e2e_faster_rcnn_R-50-FPN_giou_1x.yaml --load_ckpt
faster_giou.pth
ERROR:
RuntimeError: Error(s) in loading state_dict for Generalized_RCNN:
While copying the parameter named "Box_Head.fc1.weight", whose dimensions in the model are torch.Size([1024, 50176]) and whose dimensions in the checkpoint are torch.Size([1024, 12544]).

Then I run
python tools/train_net_step.py --dataset coco2017 --cfg configs/baselines/e2e_faster_rcnn_R-50-FPN_giou_1x.yaml --use_tfboard
and get my model(641mb),
so I confused which model is right ? 641MB vs 333MB?
I need your help,thanks

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