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

Making inference from a pretrained HRNet? #29

Open
Ainecop opened this issue Jun 22, 2023 · 1 comment
Open

Making inference from a pretrained HRNet? #29

Ainecop opened this issue Jun 22, 2023 · 1 comment

Comments

@Ainecop
Copy link

Ainecop commented Jun 22, 2023

Great Work. I was wondering How can I make inference from a pretrained HRNet. Please do provide a pretrained HRNet model or provide the link as the dimensions are mismatching when i downloaded hrnet pretrained from HRNet official.
What parameters do you set for pretrained HRNet ? Waiting for a positive response.

@Ainecop
Copy link
Author

Ainecop commented Jun 22, 2023

    size mismatch for transition1.1.0.1.running_mean: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([64]).
    size mismatch for transition1.1.0.1.running_var: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([64]).
    size mismatch for stage2.0.branches.0.0.conv1.weight: copying a param with shape torch.Size([48, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([32, 32, 1, 1]).
    size mismatch for stage2.0.branches.0.0.bn1.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current 

model is torch.Size([32]).
size mismatch for stage2.0.branches.0.0.bn1.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([32]).
size mismatch for stage2.0.branches.0.0.bn1.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([32]).
size mismatch for stage2.0.branches.0.0.bn1.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([32]).
size mismatch for stage2.0.branches.0.0.conv2.weight: copying a param with shape torch.Size([48, 48, 3, 3]) from checkpoint, the shape in current model is torch.Size([32, 32, 3, 3]).
size mismatch for stage2.0.branches.0.0.bn2.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current
model is torch.Size([32]).
size mismatch for stage2.0.branches.0.0.bn2.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([32]).
size mismatch for stage2.0.branches.0.0.bn2.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([32]).
size mismatch for stage2.0.branches.0.0.bn2.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([32]).

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