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

EVA 02 vit_h pre training weight loading problem #1899

Open
lyc200522 opened this issue Jun 3, 2024 · 0 comments
Open

EVA 02 vit_h pre training weight loading problem #1899

lyc200522 opened this issue Jun 3, 2024 · 0 comments

Comments

@lyc200522
Copy link

Branch

main branch (mmpretrain version)

Describe the bug

size mismatch for backbone.rope.freqs_sin: copying a param with shape torch.Size([16, 16, 64]) from checkpoint, the shape in current model is torch.Size([256, 64]).
missing keys in source state_dict: backbone.ln2.weight, backbone.ln2.bias

Environment

import torch
from mmpretrain import get_model
model = get_model('vit-large-p14_eva02-pre_in21k', pretrained=True)

Other information

No response

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